python - 将颜色条限制在指定范围内

Question

我想在 Matplotlib (v3.1.0) 中生成填充等高线图，并通过限制颜色范围仅突出显示狭窄范围的值。在我当前的方法中，颜色条扩展到整个值范围，我想将此范围限制为 (vmin, vmax) 并使用 extend 来指示颜色被剪裁。请参阅以下示例:

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 1, 100)
y = np.linspace(0, 1, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2

fig, ax = plt.subplots(nrows=1, ncols=1)
CS = ax.contourf(x, y, Z.T, levels=200, vmin=0.9, vmax=1.1, cmap="RdBu")
cb0 = fig.colorbar(CS, ax=ax, ticks=[0.9, 1.0, 1.1], extend="both")
plt.tight_layout()
plt.show()

正如您所看到的，颜色条不仅限于(0.9, 1.1)，也不会扩展(箭头不会出现)。有什么想法吗？

Answer 1

您需要将级别设置为您感兴趣的值范围。

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 1, 100)
y = np.linspace(0, 1, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2

fig, ax = plt.subplots(nrows=1, ncols=1)
CS = ax.contourf(x, y, Z.T, levels=np.linspace(0.9,1.1,51), 
                 vmin=0.9, vmax=1.1, cmap="RdBu", extend='both')
cb0 = fig.colorbar(CS, ax=ax, ticks=[0.9, 1.0, 1.1], extend="both")
plt.tight_layout()
plt.show()