[python]随机选取的方式——random.choices()

关于随机选取的函数。

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'])

执行结果为：