我有一个数组,包括体面的观察,不相关的观察(我想掩盖),以及没有观察的区域(我也想掩盖).我希望将此数组显示为具有两个单独掩码的图像(使用pylab.imshow),其中每个掩码以不同的颜色显示.
我找到了某种颜色的单个掩码(here)的代码,但两个不同的掩码没有任何代码：
masked_array = np.ma.array (a,mask=np.isnan(a))
cmap = matplotlib.cm.jet
cmap.set_bad('w',1.)
ax.imshow(masked_array,interpolation='nearest',cmap=cmap)
如果可能的话,我想避免使用严重扭曲的颜色图,但接受这是一个选项.
最佳答案
根据某些条件,您可能只是使用某个固定值替换数组中的值.例如,如果要屏蔽大于1且小于-1的元素：
val1,val2 = 0.5,1
a[a<-1]= val1
a[a>1] = val2
ax.imshow(a,interpolation='nearest')
可以修改val1和val2以获得所需的颜色.
您还可以明确设置颜色,但需要更多工作：
import matplotlib.pyplot as plt
from matplotlib import colors,cm
a = np.random.randn(10,10)
norm = colors.normalize()
cmap = cm.hsv
a_colors = cmap(norm(a))
col1 = colors.colorConverter.to_rgba('w')
col2 = colors.colorConverter.to_rgba('k')
a_colors[a<-0.1,:] = col1
a_colors[a>0.1,:] = col2
plt.imshow(a_colors,interpolation='nearest')
plt.show()