python - 如何用两种颜色绘制imshow？

Question

这是我的代码的简单版本

m=10
n=40
z=np.random.uniform(0,1,(m,n))
plt.imshow(z,cmap='Greys')  
plt.colorbar()

<小时/>

这是输出

<小时/>

这就是我想做的

我希望每隔一列为红色，其他列为蓝色，我该怎么做？

我已经尝试过这个，但它给出了一个错误

for i in range(n):  
if (np.mod(i,2)==1):  
   plt.imshow(z[:,i],cmap='Blues')  
else :  
    plt.imshow(z[:,i],cmap='Reds')

ERROR: Invalid shape (10,) for image data

Answer 1

m=10
n=40
z=np.random.random(size=(m,n))
z1 = np.full(z.shape, np.nan)
z2 = np.full(z.shape, np.nan)
z1[:,::2] = z[:,::2]
z2[:,1::2] = z[:,1::2]
plt.imshow(z1,cmap='Reds')
plt.imshow(z2,cmap='Blues')