如何在Python中使用colorbar Matplotlib绘制一个2D矩阵？

要在Python中使用colorbar绘制2D矩阵，我们可以使用numpy创建一个2D数组矩阵，并将该矩阵用于 imshow() 方法。

步骤

• 使用numpy创建 data2D 。

• 使用 imshow() 方法将数据显示为图像，即在2D正则光栅上。

• 使用 colorbar() 方法和 imshow() 标量映射图像创建ScalarMappable实例 mappable 的颜色条。

• 使用 show() 方法显示图形。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
data2D = np.random.random((50, 50))
im = plt.imshow(data2D, cmap="copper_r")
plt.colorbar(im)
plt.show()

输出