码迷,mamicode.com
首页 > 编程语言 > 详细

Python带权重随机数的简单实现

时间:2016-05-30 16:03:39      阅读:1389      评论:0      收藏:0      [点我收藏+]

标签:python   随机数带   权重   

该方法是常用的带权重随机数生成方法,思路是先将权重值求和total,在0与权重和total之间获得一个随机数rd,遍历权重字典,累加其权重值weight_sum, 当rd小于或等于weight_sum时,返回当前的权重key值,示例代码如下:

import random

def random_weight(weight_data):
    _total = sum(weight_data.values())    # 权重求和
    _random = random.uniform(0, _total)   # 在0与权重和之前获取一个随机数 
    _curr_sum = 0
    _ret = None
    try:
        _keys = weight_data.iterkeys()    # 使用Python2.x中的iterkeys
    except AttributeError:
        _keys = weight_data.keys()        # 使用Python3.x中的keys
    for _k in _keys:
        _curr_sum += data[_k]             # 在遍历中,累加当前权重值
        if _random <= _curr_sum:          # 当随机数<=当前权重和时,返回权重key
            _ret = _k
            break
    return _ret

转入值是一个字典,key为要获得的随机数据,key为其权重,如{‘a‘: 10, ‘b‘: 15, ‘c‘: 50}

Python带权重随机数的简单实现

标签:python   随机数带   权重   

原文地址:http://techvincent.blog.51cto.com/11654785/1784327

(0)
(1)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!