python - 如何根据浮点值从字典中随机选择一个键

Question

假设我有一个字典

{'option one': 5.0, 'option two': 5.0, 'option three': 10.0}

我如何根据上述概率随机选择一个 key (即选项一和选项二将有 25% 的概率被选中。选项 3 将有 50% 的概率被选中)

Answer 1

作为单行:

import random

random.seed(100)
d = {'option one': 5.0, 'option two': 5.0, 'option three': 10.0}
picked = random.choices(*zip(*d.items()))[0]
print(picked)
# option one

更多分割:

import random

random.seed(100)
d = {'option one': 5.0, 'option two': 5.0, 'option three': 10.0}
# Key-value pairs in dictionary
items = d.items()
# "Transpose" items: from key-value pairs to sequence of keys and sequence of values
values, weights = zip(*items)
# Weighted choice (of one element)
picked = random.choices(values, weights)[0]
print(picked)
# option one

注意random.choices (与 random.choice 不同，它提供了一个 weights 参数)是在 Python 3.6 上添加的。