如何为 Matplolib imshow 绘图颜色条添加标签?

matplotlibpythondata visualization

要为 matplotlib imshow() 绘图颜色条添加标签,我们可以采取以下步骤 −

  • 设置图形大小并调整子图之间和周围的填充。

  • 使用 Numpy 创建 5×5 个数据点。

  • 使用 imshow() 方法将数据显示为图像,即在 2D 常规栅格上。

  • ScalarMappable 实例 im 创建颜色条。

  • 使用 set_label() 方法设置颜色条标签。

  • 要显示图形,请使用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.random.rand(5, 5)

im = plt.imshow(data, cmap="copper")

cbar = plt.colorbar(im)
cbar.set_label("Colorbar")

plt.show()

输出


相关文章