我试图以图形方式分析2d数据。 matplotlib.imshow
非常有用,但我觉得如果我可以从我的矩阵中排除一些细胞,我可以更多地使用它,超出一系列感兴趣的值。我的问题是这些价值观会变得“平坦”。我感兴趣的色域图。排除这些值后,我可以获得更多的颜色分辨率。
我知道如何在矩阵上应用蒙版来排除这些值,但是在应用蒙版后它会返回1d对象:
mask = (myMatrix > lowerBound) & (myMatrix < upperBound)
myMatrix = myMatrix[mask] #returns a 1d array :(
有没有办法将掩码传递给imshow
如何重建2d数组?
答案 0 :(得分:7)
您可以使用numpy.ma.mask_where
来保留数组形状,例如
import numpy as np
import matplotlib.pyplot as plt
lowerBound = 0.25
upperBound = 0.75
myMatrix = np.random.rand(100,100)
myMatrix =np.ma.masked_where((lowerBound < myMatrix) &
(myMatrix < upperBound), myMatrix)
fig,axs=plt.subplots(2,1)
#Plot without mask
axs[0].imshow(myMatrix.data)
#Default is to apply mask
axs[1].imshow(myMatrix)
plt.show()