在Python中创建具有概率的二进制随机矩阵 [英] create binary random matrix with probability in 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文档。

这篇关于在Python中创建具有概率的二进制随机矩阵的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！