Python matplotlib使用colormap更改轮廓曲线图中指定值的颜色

Question

我正在尝试使用colormap在matplotlib中创建填充等高线图。 我想更改指定值的颜色。

例如

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

我希望颜色介于-1和1之间的白色保留其他颜色的默认颜色映射。

对不起，我英语不好。 任何帮助都将不胜感激。

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关键字存在的原因。