[英]In matplotlib imshow() how can I change the color axis range on an existing plot?
[英]How can I plot NaN values as a special color with imshow in matplotlib?
我试图在matplotlib中使用imshow将数据绘制为热图,但是一些值是NaN。 我希望将NaN渲染为色彩映射中找不到的特殊颜色。
例:
import numpy as np
import matplotlib.pyplot as plt
f = plt.figure()
ax = f.add_subplot(111)
a = np.arange(25).reshape((5,5)).astype(float)
a[3,:] = np.nan
ax.imshow(a, interpolation='nearest')
f.canvas.draw()
结果图像出乎意料地全是蓝色(喷射色图中的最低颜色)。 但是,如果我这样做的情节:
ax.imshow(a, interpolation='nearest', vmin=0, vmax=24)
- 然后我得到了更好的东西,但是NaN值被绘制成与vmin相同的颜色......是否有一种优雅的方式可以设置NaNs用特殊颜色绘制(例如:灰色或透明)?
嗯,似乎我可以使用蒙面数组来做到这一点:
masked_array = np.ma.array (a, mask=np.isnan(a))
cmap = matplotlib.cm.jet
cmap.set_bad('white',1.)
ax.imshow(masked_array, interpolation='nearest', cmap=cmap)
这应该足够了,尽管我仍然愿意接受建议。 :]
对于较新版本的Matplotlib,不再需要使用屏蔽数组。
例如,让我们生成一个数组,其中每第7个值都是NaN:
arr = np.arange(100, dtype=float).reshape(10, 10)
arr[~(arr % 7).astype(bool)] = np.nan
我们可以修改当前的colormap并使用以下行绘制数组:
current_cmap = matplotlib.cm.get_cmap()
current_cmap.set_bad(color='red')
plt.imshow(arr)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.