np.random.choice - 不能指定硬幣翻轉次數

Question

我試圖產生 3 次偏置硬幣翻轉的一百萬個結果 - random.randint 允許我指定翻轉次數，但 random.choice 似乎只讓我指定大小。 我想要得到的是：每次翻轉 3 次的百萬結果，硬幣正面朝上的幾率為 60%。

Answer 1

好像你想要這個：

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

或者

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