python - 你如何从python中的概率列表中采样

Question

如何从 python 中加起来为 1 的概率列表中进行采样。

前任。List: [0.1, 0.8, 0.1] 第一个元素将被选择 10% 的时间，第二个 80%，第三个 10%

标签： pythonstatistics

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相同数量的要选择的项目。