生成随机数组，其中每个条目只能属于一组离散值

Question

我想生成一个形状为(20, 30, 3)的随机矩阵。每个条目只能是255或0。如何使用numpy生成这种矩阵？

Answer 1

我们可以使用：

numpy.random.choice(a, size=None, replace=True, p=None)

a：可能值的数组

尺寸：输出形状

p：与

相关的概率

source

因此，为了解决我们的情况，每个0和255的概率为50％：

numpy.random.choice([0,255], size=(20,30,3), replace=True, p=[0.5,0.5])

Answer 2

为了提高头部或尾部的效率：

a=255*np.random.randint(0,2,(20,30,3))