如何利用Python随机从list中挑选一个元素

共计 1959 个字符，预计需要花费 5 分钟才能阅读完成。

在本文中，我们将研究从列表中选择随机元素的不同实现方法。在日常项目中，我们经常会遇到这种情形，比如随机从多种数据增强策略中选择一种或几种来提升训练数据的多样性。
闲话少说，我们直接开始吧。

为了方便示例，这里我们假设有一个包含多个水果英文名的列表，并且我们需要从中选择一种随机水果进行输出。

水果名称列表如下所示：

fruit_name_list = [
'apple',
'orange',
'banana',
'pear',
'watermelon'
]

random 库是一个 python 的内置库，因此我们不需要重新安装它，可以方便地直接导入。这里我们将研究使用该库从列表中选择随机元素的 3 种不同方法。

函数​​random.randrange(num_items)​​用于返回介于区间 0 到​​num_items-1​​之间的一个随机数，如果我们得到一个列表的随机下标，进而我们就可以方便的得到该列表中的一个随机元素。

样例代码如下：

import random
num_items = len(fruit_name_list)
random_index = random.randrange(num_items)
winner = fruit_name_list[random_index]
print(winner)

输出如下：

orange

函数​​random.choice​​ 将列表作为输入参数，并从列表中返回一个随机元素。在上述的例子中，我们可以直接传递 列表​​fruit_name_list​​作为该函数的参数。

样例代码如下：

winner = random.choice(fruit_name_list)
print(winner)

输出如下：

watermelon

函数​​random.sample​​功能类似于函数​​random.choice​​，主要区别在于我们可以指定所需的随机元素的数量。在下面的代码样例中，我们可以得到两个随机的水果英文名字。​​random.sample​​ 返回值为一个列表。

样例代码如下：

winners = random.sample(fruit_name_list,2)
print(winners)

输出如下：

['banana', 'watermelon']

​​Secrets​​ 库优于 Random 库，因为它更安全。与 random 库一样，它也是一个内置的 python 库，我们不必安装任何其他依赖项。但是，如果我们使用的是低于 3.6 的 Python 版本，则必须使用 pip 安装该模块。

函数​​secrets.randbelow​​的功能类似于​​random.randrange​​。使用该函数我们可以得到一个介于 0 和 ​​num_items – 1​​ 之间的随机索引，进而使用该索引来方便地访问我们的水果英文名列表中的元素。

样例代码如下：

import secrets
num_items = len(fruit_name_list)
random_index = secrets.randbelow(num_items)
winner = fruit_name_list[random_index]
print(winner)

输出如下：

watermelon

函数​​secrets.choice​​功能类似于函数​​random.choice​​, 该函数将列表作为输入参数，并从该列表中返回一个随机元素。

样例代码如下：

winner = secrets.choice(fruit_name_list)
print(winner)

输出如下：

apple

函数​​secrets.SystemRandom().sample()​​功能类似于 ​​random.sample()​​, 该函数接受输入列表和返回元素个数作为输入参数，同时该函数返回一个挑选出的随机元素的列表。

样例代码如下：

winners = secrets.SystemRandom().sample(fruit_name_list,2)
print(winners)

输出如下：

['watermelon', 'orange']

本文介绍了分别使用 Random 库以及使用 Secrets 库来从列表中随机挑选一个或多个元素的不同实现方法，并均给出了代码示例。

阿里云 2 核 2G 服务器 3M 带宽 61 元 1 年，有高配

腾讯云新客低至 82 元 / 年，老客户 99 元 / 年

代金券：在阿里云专用满减优惠券