灰度图局部着色

'''
purpose:把灰度值大于0.5的点着红色
1.初始图片是单通道的
2.想完成目标要改为3通道
3.着红色点的0号通道赋值为1,1号和2号通道赋值为0，非着色点3个通道的值相等，均为初始单通道时的值。

H=np.zeros((256,256,3)).astype('float32')
u>0.5

P=[[0.3,0.4],[0.8,0.9]]
P=np.asarray(P)
H[:,:,0]=(P<u)*P+(P>u)
H[:,:,1]=(P<u)*P      
H[:,:,2]=(P<u)*P
#print(H)

plt.imshow(H)
plt.axis('off')
plt.show()