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

iutyhrg

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

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

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}