设置 Matplotlib 中等值线图的色标范围

如需设定 Matplotlib 中某等值线图的色标范围，可采取以下步骤：

  • 设置图形大小并调整子图之间和周围的空白。
  • 使用 numpy 创建 x 和 y 数据点。
  • 使用 x 和 y 获取数据。
  • 从坐标向量获取坐标矩阵。
  • 设置 vmin 和 vmax 以设定 Matplotlib 中等值线图的色标范围。
  • 使用 contourf() 方法绘制等值线。
  • 在 vmin 和 vmax 范围内使用标量映射器制作色标。
  • 使用 show() 方法显示图形。

例子

from matplotlib import pyplot as plt
import numpy as np
from matplotlib.cm import ScalarMappable

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

x = np.arange(20)
y = np.arange(20)

data = x[:, None] + y[None, :]
X, Y = np.meshgrid(x, y)

vmin = 0
vmax = 15

fig, ax = plt.subplots()

qcs = ax.contourf(
      X, Y, data,
      vmin=vmin, vmax=vmax
)

fig.colorbar(
   ScalarMappable(norm=qcs.norm, cmap=qcs.cmap),
   ticks=range(vmin, vmax+5, 5)
)

plt.show()

输出

更新于：2021 年 6 月 9 日

11000+ 次浏览

开启您的 职业生涯

完成课程获取认证

开始
广告