如何在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()

输出