如何在 Matplotlib 中显示矩阵值和颜色图?

matplotlibpythondata visualization更新于 2025/9/12 16:12:17

要在 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()