如何使用 Matplotlib 中的 pylab.imshow() 显示 np.array?

matplotlibpythondata visualization更新于 2025/6/2 16:12:17

要使用 imshow() 显示 np.array，我们可以采取以下步骤

设置图形大小并调整子图之间和周围的填充。
使用 np.array 制作 2D 数据栅格。
将数据显示为图像，即在 2D 常规栅格上。
要显示图形，请使用 show() 方法。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.array(
[[0.1, 0.7, 0.6, 0.3],
[0.2, 0.6, 0.5, 0.2],
[0.8, 0.3, 0.80, 0.01],
[0.3, 0.4, 0.2, 0.1]]
)
plt.imshow(data, interpolation="nearest", cmap="RdYlGn_r")
plt.show()