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

这篇文章主要介绍了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) # 历史 # 科学 # 科学 # 科学 # 科学 # 科学 # 人文 # 科学 # 历史 # 科学

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

以上就是Python在字典中获取带权重的随机值实现方式的详细内容,更多请关注0133技术站其它相关文章!

赞(0) 打赏
未经允许不得转载:0133技术站首页 » python