如何在 Matplotlib 中显示矩阵值和颜色图?
matplotlibpythondata visualization
要在 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()
输出
