Python-- 有权重的随机选择， Weighted Random Choice

最新推荐文章于 2024-06-04 11:19:23 发布

fxsjy

最新推荐文章于 2024-06-04 11:19:23 发布

阅读量4.4k

点赞数

文章标签： random tuples function import pair list

本文链接：https://blog.csdn.net/fxsjy/article/details/5163993

版权

import random

def windex(lst):
    '''an attempt to make a random.choose() function that makes weighted choices
    accepts a list of tuples with the item and probability as a pair'''
    wtotal = sum([x[1] for x in lst])
    n = random.uniform(0, wtotal)
    for item, weight in lst:
        if n < weight:
            break
        n = n - weight
    return item