如何在 Matplotlib 中显示矩阵值和颜色映射?
要在 Matplotlib 中显示矩阵值和颜色映射,我们可以采取以下步骤:
- 设置图像大小并调整子图之间的和周围的边距。
- 创建一个图像和一组子图。
- 初始化矩阵的最大和最小值。
- 将 2D 矩阵或数组的值绘制为颜色编码的图像。
- 遍历颜色编码图像的每个单元,并将值置于中心。
- 要显示图像,请使用 show() 方法。
示例
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
fig, ax = plt.subplots()
min_val, max_val = 0, 5
matrix = np.random.randint(0, 5, size=(max_val, max_val))
ax.matshow(matrix, cmap='ocean')
for i in range(max_val):
for j in range(max_val):
c = matrix[j, i]
ax.text(i, j, str(c), va='center', ha='center')
plt.show()输出


广告
数据结构
网络
RDBMS
操作系统
Java
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP