关于随机选取的函数。
1. 列表随机选取
1.1. 随机等概率选取一个结果
首先我们来想象一下,现在有一个列表,要在其中随机选取一个数字,比如:
a = [1,2,3,4,5]
这里我们需要用到一种比较简单的随机选取方式,即random.choices
代码如下:
import random
a = [1,2,3,4,5]
num_list = random.choices(a)
print(num_list)
执行效果如下:
1.2. 随机等概率选取多个结果
注意到random.choices()的返回值默认是一个列表(即使其默认选取一个值)
如果只要一个值的话,可以再加一步处理:
num = random.choices(a)[0] # 选取列表的第一个元素
但是,如果需要选取多个结果,则可以使用参数k指定
import random
a = [1,2,3,4,5]
num_list = random.choices(a, k=2) # 选取两个
print(num_list)
结果如下:
1.3. 随机非等概率选取结果
有的时候,我们希望选取的概率并不是完全一样的,比如说,我希望a中,有1/15的概率选到1,2/15的概率选到2,以此类推,这时就需要指定参数weights设定权重
即:
import random
from collections import Counter
a = [1,2,3,4,5]
# 使用weights指定,weights接受一个列表作为参数
num_list = random.choices(a, k=1500, weights=[i/sum(a) for i in a])
# 统计每一项被选取的次数
counts = Counter(num_list)
print("1:", counts[1])
print("2:", counts[2])
print("3:", counts[3])
print("4:", counts[4])
print("5:", counts[5])
执行结果如下:
2. 字典随机选取
2.1. 随机等概率选取
随机等概率选取字典的key,其方式与列表类似,只不过需要一个list()
import random
b = {'a':1, 'b':2, 'c':3, 'd':4, 'e':5}
# 在b的key中选取一个
num_list = random.choices(list(b.keys()))
print(num_list)
其结果为:
2.2. 按照value的概率选取
既然是字典,我们也会想到使用value指定概率,其实也比较简单,使用到了weights参数
import random
from collections import Counter
b = {'a':1, 'b':2, 'c':3, 'd':4, 'e':5}
num_list = random.choices(list(b.keys()), k=1500, weights=list(b.values()))
# 统计每一项被选取的次数
counts = Counter(num_list)
print("a:", counts['a'])
print("b:", counts['b'])
print("c:", counts['c'])
print("d:", counts['d'])
print("e:", counts['e'])
执行结果为: