我怎样才能绘制NaN值作为一种特殊的颜色与matplotlib imshow?
-
24-09-2019 - |
题
我想在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()
将所得的图像是出乎意料的所有蓝色(在喷射colormap中最低的颜色)。但是,如果我做的绘图这样的:
ax.imshow(a, interpolation='nearest', vmin=0, vmax=24)
- 然后我得到更好的东西,但NaN值绘制的颜色相同VMIN ......有没有一种优雅的方式,我可以设置的NaN用一种特殊的颜色绘制(例如:灰色或透明的)?
解决方案
HRM,它出现我可以使用掩蔽阵列做到这一点:
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的较新版本,则不必再使用一个掩蔽阵列。
例如,让我们产生阵列,其中每一个第七值是NaN:
arr = np.arange(100, dtype=float).reshape(10, 10)
arr[~(arr % 7).astype(bool)] = np.nan
我们可以修改当前的颜色表,并用下面的行绘制数组:
current_cmap = matplotlib.cm.get_cmap()
current_cmap.set_bad(color='red')
plt.imshow(arr)
不隶属于 StackOverflow