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

输出

更新于：2021 年 8 月 4 日

4K+ 浏览

开启你的职业

完成课程以获得认证

开始
广告