np.random.choice - 不能指定硬币翻转次数

Question

I'm trying to produce a million results of 3 bias coin flips - random.randint allows me to specify the number of flips but random.choice seems to only let me specify the size. What's I'm trying to get is: million results of 3 flips each with 60% chance of heads on a coin.

Answer 1

Seems like you want this:

np.random.choice([0, 1], size = (10**6, 3), p = [.4, .6])

or

(np.random.rand(10**6, 3) < .6).astype(int)