Random generate element in numpy with probability

Question

I need to generate an array in numpy (there are N numbers).

There are only two kinds of element in this array, for example: 3.0 and -3.0. The probability of occurring 3.0 is 0.4, and the probability of occurring -3.0 is 0.6.

How to generate such an array?

https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.random.choice.html — user2357112, Jul 13 '17 at 21:30

score 4 · Accepted Answer · answered Jul 13 '17 at 21:34

4

This will do the job:

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

answered Jul 13 '17 at 21:34

Miriam Farber

18,986
14
61
76

Random generate element in numpy with probability

1 Answers1