python - 根据概率将变量设置为不同的值

Question

我有一个变量 x，我想将其设置为值 1 或 2，x=1 的概率为 60%，x=2 的概率为 40%。

我在课堂上这样做，所以每次单击按钮时 x 都应该重新生成，然后我会绘制它，但我不确定如何使用特定概率来设置它。我知道我可以随意做，但这不是我想要的。

有谁知道如何做到这一点？

Answer 1

Random 使用均匀分布，因此您可以像这样使用它：

x = 1 if random.random() < 0.6 else 2

Answer 2

您只需取一个随机值，检查它是否小于期望的概率，然后返回第一个或第二个选项：

def random_choice(val1, val2, probability_of_val1):
    return val1 if random.random() < probability_of_val1 else val2

Answer 3

一个简单的策略可能如下：

1. 生成0-1之间的随机数
2. 如果随机数大于 0.6，则赋值 2
3. 否则赋值 1

执行此操作的 Python 代码：

import random
if random.random() > 0.6:
  x = 2
else:
  x = 1