您的位置:首页 > 其它

按概率随机选取

2017-06-17 12:21 253 查看
上一篇笔记的pygame游戏对敌人和白云的移动速度使用了随机函数
randint()
,游戏体验不是太好。如果是按概率随机选取设置速度的话,游戏体验会好一些。

据我了解,
random.choice(seq)
是等概率选取一个,不是我想要的。而
numpy.random.choice(seq, p, k)
是按概率随机重复选取多个,这正是我想要的。

但是,我不想为这么一个函数引入巨大的numpy库,所以打算自己实现一个按概率随机选取的函数。

特此将代码记录如下:

# 作者:hhh5460
# 时间:2017年6月17日

import random

# 根据概率随机选取
def random_choice(seq, prob, k=1):
'''
功能:
按给定概率prob,从seq中选取元素。可重复k次
注意
1. seq, prob长度要相等
2. prob的概率和要等于1
3. k 表示重复选取的次数,默认为1次
4. 结果返回list
5. 用到了random模块的random()函数
例子:
>>> random_choice(['a','b','c','d'], [0.4, 0.15, 0.1, 0.35])
['d']
>>> random_choice('abcd', [0.4, 0.15, 0.1, 0.35], k=5)
['d','d','b','a','d']
'''
res = []
for j in range(k):
p = random.random()
for i in range(len(seq)):
if sum(prob[:i]) < p <= sum(prob[:i+1]):
res.append(seq[i])
return res

# 测试
def test():
print(random_choice(['a','b','c','d'], [0.4, 0.15, 0.1, 0.35], k=5))

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