Python实现基于权重的随机数2种方法
作者:junjie 时间:2022-10-13 03:25:40
问题:
例如我们要选从不同省份选取一个号码,每个省份的权重不一样,直接选随机数肯定是不行的了,就需要一个模型来解决这个问题。
简化成下面的问题:
字典的key代表是省份,value代表的是权重,我们现在需要一个函数,每次基于权重选择一个省份出来
{"A":2, "B":2, "C":4, "D":10, "E": 20}
解决:
这是能想到和能看到的最多的版本,不知道还没有更高效好用的算法。
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#python2.7x
#random_weight.py
#author: orangleliu@gmail.com 2014-10-11
'''''
每个元素都有权重,然后根据权重随机取值
输入 {"A":2, "B":2, "C":4, "D":10, "E": 20}
输出一个值
'''
import random
import collections as coll
data = {"A":2, "B":2, "C":4, "D":6, "E": 11}
#第一种 根据元素权重值 "A"*2 ..等,把每个元素取权重个元素放到一个数组中,然后最数组下标取随机数得到权重
def list_method():
all_data = []
for v, w in data.items():
temp = []
for i in range(w):
temp.append(v)
all_data.extend(temp)
n = random.randint(0,len(all_data)-1)
return all_data[n]
#第二种 也是要计算出权重总和,取出一个随机数,遍历所有元素,把权重相加sum,当sum大于等于随机数字的时候停止,取出当前的元组
def iter_method():
total = sum(data.values())
rad = random.randint(1,total)
cur_total = 0
res = ""
for k, v in data.items():
cur_total += v
if rad<= cur_total:
res = k
break
return res
def test(method):
dict_num = coll.defaultdict(int)
for i in range(100):
dict_num[eval(method)] += 1
for i,j in dict_num.items():
print i, j
if __name__ == "__main__":
test("list_method()")
print "-"*50
test("iter_method()")
一次执行的结果
A 4
C 14
B 7
E 44
D 31
--------------------------------------------------
A 8
C 16
B 6
E 43
D 27
思路:
思路都很原始可以参考下面的连接,还有别的好方法一起交流!!
代码: https://gist.github.com/orangle/d83bec8984d0b4293710
参考:
https://www.jb51.net/article/65060.htm
https://www.jb51.net/article/65058.htm
标签:Python,权重,随机数
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
CSS hack:区分IE6,IE7,firefox
2007-12-23 10:25:00
PHP中非常有用却鲜有人知的函数集锦
2023-11-24 14:29:47
c++生成dll使用python调用dll的方法
2023-02-16 11:40:05
重构Python代码的六个实例
2023-08-07 02:10:14
MySQL常用维护管理工具
2009-03-08 16:51:00
![](https://img.aspxhome.com/file/UploadPic/20093/8/phpmyadmin-39s.jpg)
Python中的“_args”和“__kwargs”用法详解
2023-03-11 02:10:38
深入浅析python的第三方库pandas
2021-06-05 03:13:03
![](https://img.aspxhome.com/file/2023/0/69010_0s.png)
浅谈Golang内存逃逸
2023-08-27 07:51:55
SqlServer 索引自动优化工具
2012-10-07 10:41:09
实例剖析:MySQL数据库优化详解
2008-11-22 12:19:00
JavaScript解决Joseph问题
2008-06-21 17:11:00
Monster for Chrome
2010-05-04 16:30:00
![](https://img.aspxhome.com/file/UploadPic/20105/4/monsterui-64s.png)
Python实现自动化发送邮件
2023-12-07 14:39:58
![](https://img.aspxhome.com/file/2023/7/67827_0s.jpg)
FckEditor配置手册中文教程详细说明
2010-02-28 12:37:00
解析array splice的移除数组中指定键的值,返回一个新的数组
2023-11-18 09:51:28
在Vista IIS 7 中用 vs2005 调试 Web 项目的注意事项
2011-04-18 10:38:00
![](https://img.aspxhome.com/file/UploadPic/20112/5/20112511198969s.jpg)
Python绘制词云图之可视化神器pyecharts
2023-02-23 01:35:25
![](https://img.aspxhome.com/file/2023/9/69449_0s.png)
phpstorm断点调试方法图文详解
2023-05-30 01:06:40
![](https://img.aspxhome.com/file/2023/9/55259_0s.jpg)
java EJB 加密与解密原理的一个例子
2023-10-02 06:41:16
简单理解Python中的装饰器
2021-06-18 01:44:21