python - 如何根据概率从列表中选择项目

Question

我有清单a和b

a = [0.1, 0.3, 0.1, 0.2, 0.1, 0.1, 0.1]

b = [apple, gun, pizza, sword, pasta, chicken, elephant]

现在我想创建一个包含 3 个项目的新列表 c

这 3 个项目是根据列表 a 中的概率从列表 b 中选择的

项目不应在列表 c 中重复

例如-我正在寻找的输出

c = [gun,sword,pizza]

或者

c = [apple, pizza, pasta]

注意 （列表 a 的所有值的总和为 1，列表 a 和 b 中的项目数相同，实际上我在列表 a 和 b 中都有一千个项目，我想根据分配的概率从列表中选择一百个项目对他们来说，python3）

标签： pythonpython-3.xprobability

Answer 1

使用random.choices：

>>> import random
>>> print(random.choices(
...     ['apple', 'gun', 'pizza', 'sword', 'pasta', 'chicken', 'elephant'], 
...     [0.1, 0.3, 0.1, 0.2, 0.1, 0.1, 0.1],
...     k=3
... ))
['gun', 'pasta', 'sword']

编辑：为避免替换，您可以从总体中删除所选项目：

def choices_no_replacement(population, weights, k=1):
    population = list(population)
    weigths = list(weights)    
    result = []
    for n in range(k):
        pos = random.choices(
            range(len(population)), 
            weights,
            k=1
        )[0]
        result.append(population[pos])
        del population[pos], weights[pos]
    return result

测试：

>>> print(choices_no_replacement(
...     ['apple', 'gun', 'pizza', 'sword', 'pasta', 'chicken', 'elephant'],
...     [0.1, 0.3, 0.1, 0.2, 0.1, 0.1, 0.1],
...     k=3
... ))
['gun', 'pizza', 'sword']