关键词

详解Python random.choices(从序列中获取指定数量随机元素)函数的使用方法

Python中的random模块可以用来生成伪随机数。在这个模块中,choices()函数可以从给定的序列中随机选取元素。这个函数可以用来模拟实际场景,比如抽奖,随机分配任务等等。

choices()函数的语法如下:

random.choices(population, weights=None, *, cum_weights=None, k=1)

其中:

  • population:必须是一个序列,可以是列表,元组,字符串等等。表示要从中选取元素的序列。
  • weights:可选参数。必须是一个数字序列,长度必须和population相同。表示每个元素被选中的概率,可以是小数,但必须大于等于0。
  • cum_weights:可选参数。必须是一个数字序列,长度必须和population相同。表示累计权重,即前n个元素被选中的概率,如果指定了cum_weights,那么不能同时指定weights。
  • k:可选参数。表示要选取的元素个数,默认值是1。

当参数中只指定population时,函数会从中等概率地选取1个元素,并返回一个列表。下面是一个简单的例子:

import random

# 从1-10中随机选取一个数
num = random.choices(range(1, 11))
print(num)

运行结果可能是:

[4]

当然,你也可以指定元素的权重来控制被选中的概率。例如:

import random

# 从1-10中随机选取一个数,2被选中的概率是1/3,其他数字被选中的概率是1/7
num = random.choices(range(1, 11), weights=[1, 3, 1, 1, 1, 1, 1, 1, 1, 1])
print(num)

运行结果可能是:

[2]

除了从序列中选取元素外,choices()函数还可以用来做随机采样和随机洗牌等操作。接下来给出两个实例。

实例1:使用choices()函数实现随机采样

假设我们有一个大的数据集,我们想从中随机选取一部分数据进行训练,一部分数据进行测试。这时可以使用choices()函数来实现随机采样。下面是一个例子,首先生成100个数据,然后随机选取10个数据作为测试集。

import random

data = list(range(0, 100))
test_set = random.choices(data, k=10)
train_set = [x for x in data if x not in test_set]

print("测试集:", test_set)
print("训练集:", train_set)

运行结果可能是:

测试集: [78, 91, 70, 12, 56, 11, 16, 10, 49, 61]
训练集: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 50, 51, 52, 53, 54, 55, 57, 58, 59, 60, 62, 63, 64, 65, 66, 67, 68, 69, 71, 72, 73, 74, 75, 76, 77, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 92, 93, 94, 95, 96, 97, 98, 99]

实例2:使用choices()函数实现随机洗牌

假设我们有一副扑克牌,我们想把它洗牌后重新排列。这时可以使用choices()函数来实现随机洗牌。下面是一个例子,首先生成一副扑克牌,然后随机洗牌。

import random

# 生成一副扑克牌
suits = ['♠', '♥', '♦', '♣']
ranks = ['2', '3', '4', '5', '6', '7', '8', '9', '10', 'J', 'Q', 'K', 'A']
deck = [rank + suit for suit in suits for rank in ranks]

# 洗牌
random.shuffle(deck)

print(deck)

运行结果可能是:

['7♦', 'K♦', 'A♣', 'A♦', 'J♣', '4♦', '9♥', 'K♥', '2♥', '2♠', 'K♣', '4♠', '8♦', '5♣', 'J♠', '5♠', 'Q♦', 'Q♠', '6♣', '10♠', '7♠', '9♣', 'A♥', '6♥', '6♠', '8♠', 'J♦', 'Q♣', '9♠', '3♣', '7♥', 'K♠', '2♣', '3♠', '10♥', '3♥', '4♥', '5♥', '8♥', '10♣']

通过以上两个例子,我们可以看到choices()函数的用法非常灵活,可以用来解决很多实际问题。

本文链接:http://task.lmcjl.com/news/3977.html

展开阅读全文