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

Question

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

Answer 1

我们可以使用：

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

一个：与

source

因此相关联的概率：输出形状

号码：可能值

大小的数组，用0和255的概率来解决我们的情况每个：

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))