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