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

Python 生成随机数组,其中每个条目只能属于一组离散值,python,numpy,scipy,Python,Numpy,Scipy,我想生成一个随机矩阵,形状(20,30,3)。每个条目只能是255或0。如何使用numpy生成此类矩阵?我们可以使用: numpy.random.choice(a, size=None, replace=True, p=None) a:可能值的数组 大小:输出形状 p:与一个事件相关的概率 因此,要以50%的概率(0和255)解决我们的问题: numpy.random.choice([0,255], size=(20,30,3), replace=True, p=[0.5,0.5]) 对于

我想生成一个随机矩阵,形状
(20,30,3)
。每个条目只能是
255
0
。如何使用
numpy
生成此类矩阵?

我们可以使用:

numpy.random.choice(a, size=None, replace=True, p=None)
a:可能值的数组

大小:输出形状

p:与一个事件相关的概率

因此,要以50%的概率(0和255)解决我们的问题:

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

对于正面或反面方法的效率:

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