本文介绍了在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中创建具有概率的二进制随机矩阵的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除!