Python Matplotlib如何在contourf图中使用colormap更改指定值的颜色

Question

我想使用colormap在matplotlib中创建填充的等高线图。我希望能够改变指定值的颜色。

例如，

levs = [-3,-1,1,3]
plt.contourf(x,y,z,levs,cmap=cm.jet,extend='both')

我希望将-1到1之间的颜色变为白色，保持其他颜色不变，默认使用颜色映射。

对于我的英语不好表示抱歉。 任何帮助都会受到赞赏。

- user1979961

Answer 1

我会在同一张等高线图上添加另外一个等高线图，但是只显示-1到1之间的等高线，颜色为白色。

例如：

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

x, y = np.mgrid[-100:100,-100:100]
x /= 10.
y /= 10.
r = np.sqrt(x*x+y*y)
z = 10*np.sin(r)/(r+0.01)
levels = [-4, -3, -2, -1, 1, 2, 3, 4]
plt.contourf(x, y, z, levels=levels, extend='both', cmap=cm.jet)
levels = [-1, 1]
plt.contourf(x, y, z, levels=levels, colors='w')
plt.savefig('contours.png')

更新

如果您需要更多的控制，可以在contourf中设置colors关键字为一组matplotlib颜色元组，它们的元素数量与级别相同（减1：颜色对应于级别边界之间的间隔）。然后，您就不需要重叠一个单独的轮廓线，并且您的色条也是正确的：

# same as before
levels = [-4, -3, -2, -1, 1, 2, 3, 4]
plt.contourf(x, y, z, levels=levels, extend='both', colors=('#ff0000', '#ff9900', '#999900', 'w', '#009999', '#0099ff', '#0000ff'))
plt.savefig('contours.png')

也许更好的方法是定义自己的色图，但这样就会遇到一个问题，你必须与等高线级别精确匹配。这就是为什么colors 关键字非常方便的原因。

- user707650

---

谢谢！它运行得很好。如果可能的话，我想改变plt.colorbar()绘制的颜色条的颜色列表。 - user1979961

看看我对答案所做的修改。实际上，我认为这个更新对你的问题比我的最初答案更好。 - user707650