如何利用Python隨機從list中挑選一個元素

在本文中，我們將研究從列表中選擇隨機元素的不同實現方法。在日常專案中，我們經常會遇到這種情形，比如隨機從多種資料增強策略中選擇一種或幾種來提升訓練資料的多樣性。
閒話少說，我們直接開始吧。

為了方便示例，這裡我們假設有一個包含多個水果英文名的列表，並且我們需要從中選擇一種隨機水果進行輸出。

水果名稱列表如下所示：

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庫來從列表中隨機挑選一個或多個元素的不同實現方法，並均給出了程式碼示例。