Python字典带权重随机值获取方法
发布时间：2026-10-16 |
3人浏览过
本文介绍如何在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字典权重随机值内容请搜索本站以前的文章或继续浏览下面的相关文章希望大家以后多多支持本站！
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
