如何从python中的概率列表中取样

Question

在python中，如何从加起来等于1的概率列表中取样

前。 列表：[0.1,0.8,0.1] 第一个元素将在10%的时间内选择，第二个元素为80%，第三个元素为10%

Answer 1

使用random.choices函数允许的权重。例如：

import random

floats = [0.1, 0.8, 0.1]
weights = [0.1, 0.8, 0.1]
k = 1

choice = random.choices(population=floats, weights=weights, k=k)
print(choice)

它以列表的形式返回，因为k可以等于要选择的项目数