在Python中创建具有概率的二进制随机矩阵

问题描述

我是新手。我想编写一个函数来生成一个随机nxm二进制矩阵，概率p的每个值为0。

我做了什么。

def randbin(M,N,P):              # function to generate random binary matrix    
    mat = (np.random.rand(M,N)>=P).astype(int)
    return mat  

y = randbin(5,4,0.3)

每次打印输出时，都没有按照估计的概率得到结果。我不知道我做错了什么。

推荐答案

我看不出您的方法有问题...生成概率P为0的0和1的随机矩阵的更好方法是使用random.choice：

def randbin(M,N,P):  
    return np.random.choice([0, 1], size=(M,N), p=[P, 1-P])

为了更好地理解，请查看random.choice文档。