随机生成具有概率的numpy元素

时间:2017-07-13 21:28:03

标签: python numpy

我需要在numpy中生成一个数组(有N个数字)。

此数组中只有两种元素,例如:3.0和-3.0。发生3.0的概率为0.4,发生-3.0的概率为0.6。

如何生成这样的数组?

1 个答案:

答案 0 :(得分:3)

这将完成这项工作:

n=10
np.random.choice([3,-3],n,p=[0.4,0.6])