您的位置:首页 > 编程语言 > Python开发

004_021 Python 以指定的概率获取元素 以一个列表为基准概率,从一个列表中随机获取元素

2014-03-31 22:40 344 查看
代码如下:

#encoding=utf-8

print '中国'

#以指定的概率获取元素 以一个列表为基准概率,从一个列表中随机获取元素

import random

def random_pick(some_list, probabilities):
x = random.uniform(0,1)
cumulative_probability = 0.0
for item, item_probability in zip(some_list, probabilities):
cumulative_probability += item_probability
if x < cumulative_probability:break
return item

some_list = [1,2,3,4]
probabilities = [0.2,0.1,0.6,0.1]

print random_pick(some_list,probabilities)

#根据权重来获取 核心在于权重乘以 就相当于次数

def random_pick_odd(some_list, odds):
print 'table2'
table = [z for x,y in zip(some_list,odds) for z in [x] * y]
print table
return random.choice(table)

some_list = [1,2,3,4]
odds=[3,1,4,2]

print random_pick_odd(some_list,odds)

#for item in random_picks()


打印结果如下:

中国

1

table2

[1, 1, 1, 2, 3, 3, 3, 3, 4, 4]

4
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐