如何使用 Matplotlib 在 Python 中绘制带颜色条的 2D 矩阵?
为了在 Python 中绘图带颜色条的 2D 矩阵,我们可以使用 numpy 来创建一个 2D 数组矩阵并在 imshow() 方法中使用那个矩阵。
步骤
使用 numpy 创建 data2D。
使用 imshow() 方法将数据显示为图像,即显示在 2D 正则光栅上。
使用 colorbar() 方法和 imshow() 标量可映射图像为 *图例*(ScalarMappable instance)创建颜色条。
使用 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()
输出
广告