在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()