如何添加 Matplotlib Colorbar 刻度?

matplotlibpythondata visualization

要向颜色栏添加刻度,我们可以采取以下步骤 −

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

  • 使用 numpy 创建 x、yz 数据点。

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

  • 使用 numpy 在 zminma​​x 范围内创建刻度。

  • ScalarMappable 实例创建颜色栏, *mappable*,其中 ticks=ticks

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

示例

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x, y = np.mgrid[-1:1:100j, -1:1:100j]
z = (x + y) * np.exp(-5.0 * (x ** 2 + y ** 2))

plt.imshow(z, extent=[-1, 1, -1, 1])

ticks = np.linspace(z.min(), z.max(), 5, endpoint=True)

cb = plt.colorbar(ticks=ticks)

plt.show()

输出


相关文章