单通道逻辑数组，变三通道

1 2	import numpy as np mask = np.repeat(mask[..., np.newaxis], repeats=3, axis=2)

np.newaxis 在数组哪一维出现，就默认在该处新增一个维度;
... 在numpy中可以代替 :（第一个 ... 可以代替任意多个 :，后面的每一个 ... 一个只能代替一个 :）, 以保证推导的维度数目与变量本身维度一致

imageio 读取与保存图片

import numpy as np
img = np.random.rand(512,512,3)
c = np.array([r,g,b])
np.all(img==c,axis=-1)