加权选择简短

Question

如果我在列表中有一组项目.我想根据另一个权重列表从该列表中进行选择.

例如我的收藏是['one', 'two', 'three']和权重[0.2, 0.3, 0.5],我希望这个方法在所有抽奖的大约一半中给我'三'.

最简单的方法是什么？

Answer 1

从numpy版本1.7开始,您可以使用numpy.random.choice():

elements = ['one', 'two', 'three'] 
weights = [0.2, 0.3, 0.5]

from numpy.random import choice
print(choice(elements, p=weights))

Answer 2

从Python 3.6开始,您可以使用加权随机选择(替换)random.choices.

随机.选择(人口,权重=无,*,cum_weights =无,k = 1)

用法示例:

import random
random.choices(['one', 'two', 'three'], [0.2, 0.3, 0.5], k=10)
# ['three', 'two', 'three', 'three', 'three',
#  'three', 'three', 'two', 'two', 'one']

Answer 3

此函数有两个参数:权重列表和包含可供选择对象的列表:

from numpy import cumsum
from numpy.random import rand
def weightedChoice(weights, objects):
    """Return a random item from objects, with the weighting defined by weights 
    (which must sum to 1)."""
    cs = cumsum(weights) #An array of the weights, cumulatively summed.
    idx = sum(cs < rand()) #Find the index of the first weight over a random value.
    return objects[idx]

它不使用任何python循环.

Answer 4

您可以使用多项分布（来自 numpy）来做您想做的事。例如

elements = ['one', 'two', 'three'] 
weights = [0.2, 0.3, 0.5]


import numpy as np

indices = np.random.multinomial( 100, weights, 1)
#=> array([[20, 32, 48]]), YMMV

results = [] #A list of the original items, repeated the correct number of times.
for i, count in enumerate(indices[0]):
    results.extend( [elements[i]]*count )

所以第一个位置的元素出现了 20 次，第二个位置的元素出现了 32 次，第三个位置的元素出现了 48 次，这大致是给定权重的预期。

如果您很难理解多项式分布，我发现文档真的很有帮助。