如何在 python 中生成具有给定概率的随机整数？

Question

我想根据给定的概率给 output 一个随机标志（0,1 或 2）。

例如

flag = gen_flag(p0,p1,p2)

其中p0+p1+p2=1和p0 、 p1和p2分别表示生成标志 0、1 和 2 的概率。 例如， gen_flag(0.8,0.1,0.1)很可能 output 为零。

我怎么做？

Answer 1

使用随机选择

from random import choices

flag = choices([0,1,2], [p0, p1, p2])[0]

p0等人。 不必总和为 1； 如果它们还没有标准化，它们就会被标准化。

即使在从可能的标志集中仅选择一个元素的默认情况下， choices总是返回一个列表。

Answer 2

@Chepner 有最好的答案，但直到 python 3.6 才随机添加选项。 对于早期版本，我将获取输入并制作一个 100 项列表，其中可能的数量分别为 0、1 和 2。 然后使用随机模块选择一个项目。

import random

gen_flag(p0, p1, p2):
    myList = [0]*int(p0*100) + [1]*int(p1*100) + [2]*int(p2*100)
    return myList[random.randint(0,99)]