Python在字典中获取带权重的随机值实现方式

一、前言

python在数组中随机取值有现成的方法，但是要给每个随机值被取到的概率加权重的话，可以参考下面这个方法

二、实现方式

import random

d编程客栈ef rand编程客栈om_with_weight(data_dict):
    s开发者_Python入门um_wt = sum(data_dict.values())     # 计算权重和 sum_wt 
    ra_wt = random.uniform(0, sum_wt)    # 随机获取 0-sum_wt 之间的一个浮点数 ra_wt 
    cur_wt = 0
    for key in data_dict.keys():
        cur_wt += data_dict[key]        # 遍历并累加当前权重值
        if ra_wt <= cur_wt:             # 当随机数 <= 当前权重和时，返回权重对应的key
            return key


if __name__ == '__main__':
    # 权重字典
    data_dict = {'历史': 1, 编程客栈'人文': 2, '科学': 8}
    # 取十次测试结果
    for _ in ran编程ge(10):
        value = random_with_weight(data_dict)
        print(value)
        # 历史
        # 科学
        # 科学
        # 科学
        # 科学
        # 科学
        # 人文
        # 科学
        # 历史
        # 科学

到此这篇关于Python在字典中获取带权重的随机值的文章就介android绍到这了,更多相关Python字典权重随机值内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们！