python带权重的列表中随机取值

python带权重的随机取值

代码如下，非常好用

import random
def random_weight(weight_data):
    total = sum(weight_data.values())    # 权重求和
    ra = random.uniform(0, total)   # 在0与权重和之前获取一个随机数 
    curr_sum = 0
    ret = None
    keys = weight_data.iterkeys()    # 使用Python2.x中的iterkeys
#     keys = weight_data.keys()        # 使用Python3.x中的keys
    for k in keys:
        curr_sum += weight_data[k]             # 在遍历中，累加当前权重值
        if ra <= curr_sum:          # 当随机数<=当前权重和时，返回权重key
            ret = k
            break
    return ret
weight_data = {'a': 10, 'b': 15, 'c': 50}
random_weight(weight_data)

sfk8帝国模板网测试过