网络编程
位置:首页>> 网络编程>> Python编程>> Python在字典中获取带权重的随机值实现方式

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

作者:什么都干的派森  发布时间:2022-12-11 05:21:44 

标签:Python,字典,权重,随机值

一、前言

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

二、实现方式

import random

def random_with_weight(data_dict):
   sum_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 range(10):
       value = random_with_weight(data_dict)
       print(value)
       # 历史
       # 科学
       # 科学
       # 科学
       # 科学
       # 科学
       # 人文
       # 科学
       # 历史
       # 科学

来源:https://blog.csdn.net/weixin_43721000/article/details/128109870

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com