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

Question

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

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

Answer 1

我看不出你的方法有什么问题...生成概率 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 文档。