问题1. 有一堆数据 x = [1,5,7],随机取出一个数?
random.choice(x)
问题2. 如何按照权重20 , 20 , 60取出一个数?
我们可以把总和 100分成几个区间,0 - 19 分给 1, 20 - 39 分给 5, 40 - 99 分给7, 再随机0 - 99 看这个随机数落在哪个区间就可以了。
import random
import bisect
m = [20, 20, 60]
n = []
count = 0
for i, v in enumerate(m):
n.append(count)
count += v
rd = random.randint(0, count -1)
r = bisect.bisect_right(n, rd)
result = x[r]
注意:m必须是排好顺序的,bisect二分查找binary search需要 bisect.bisect_right(n, rd) 插入的 rd 在 n 的位置。
其中 n 的值可以使用
from itertools import accumulate
n = list(accumulate(data))
使用 Python 写算法的时候,注意多找找相关的第三方包,总有惊喜等着你。
网友评论