python - 使用 contourf 设置颜色条范围

Question

这似乎是这个问题的重复:

Set Colorbar Range in matplotlib

本质上我想设置 colorbar 范围来设置限制，例如0 到 2。当我使用 vmin 和 vmax 时，contourf 中的颜色范围设置正确，但 colorbar 仅显示裁剪范围，即链接中的解决方案在使用 contourf 时似乎不起作用。我是否遗漏了一些明显的东西？

import numpy as np
import matplotlib.pyplot as plt
fld=np.random.rand(10,10)
img=plt.contourf(fld,20,cmap='coolwarm',vmin=0,vmax=2)
plt.colorbar(img)

导致

如何使用 contourf 强制颜色条范围为 0 到 2？

Answer 1

contourf 的工作方式确实与其他 ScalarMappable 有点不同。如果您指定级别数(在本例中为 20)，它们将在最小数据和最大数据之间(大约)。如果你想在两个特定值 vmin 和 vmax 之间有 n 级别，你需要将它们提供给轮廓函数

levels = np.linspace(vmin, vmax, n+1)
plt.contourf(fld,levels=levels,cmap='coolwarm')

完整代码:

import numpy as np
import matplotlib.pyplot as plt
fld=np.random.rand(10,10)
levels = np.linspace(0,2,21)
img=plt.contourf(fld,levels=levels,cmap='coolwarm')
plt.colorbar(img)
plt.show()