首頁 > 軟體

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

2022-05-24 22:04:08

1. 引言

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

2. 舉個栗子

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

水果名稱列表如下所示:

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

3. 使用Random庫

random 庫是一個 python的內建庫,因此我們不需要重新安裝它,可以方便地直接匯入。這裡我們將研究使用該庫從列表中選擇隨機元素的 3 種不同方法。

3.1 隨機下標

函數​​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

3.2 隨機選擇單個元素

函數​​random.choice​​ 將列表作為輸入引數,並從列表中返回一個隨機元素。在上述的例子中,我們可以直接傳遞 列表​​fruit_name_list​​作為該函數的引數。

樣例程式碼如下:

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

輸出如下:

watermelon

3.3 隨機選擇多個元素

函數​​random.sample​​功能類似於函數​​random.choice​​,主要區別在於我們可以指定所需的隨機元素的數量。在下面的程式碼樣例中,我們可以得到兩個隨機的水果英文名字。 ​​random.sample​​ 返回值為一個列表。

樣例程式碼如下:

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

輸出如下:

['banana', 'watermelon']

4. 使用Secrets庫

​Secrets​​ 庫優於 Random 庫,因為它更安全。與 random 庫一樣,它也是一個內建的 python 庫,我們不必安裝任何其他依賴項。但是,如果我們使用的是低於3.6 的Python版本,則必須使用pip安裝該模組。

4.1 隨機下標

函數​​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

4.2 隨機選擇單個元素

函數​​secrets.choice​​功能類似於函數​​random.choice​​,該函數將列表作為輸入引數,並從該列表中返回一個隨機元素。

樣例程式碼如下:

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

輸出如下:

apple

4.3 隨機選擇多個元素

函數​​secrets.SystemRandom().sample()​​功能類似於 ​​random.sample()​​,該函數接受輸入列表和返回元素個數作為輸入引數,同時該函數返回一個挑選出的隨機元素的列表。

樣例程式碼如下:

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

輸出如下:

['watermelon', 'orange']

5. 總結

本文介紹了分別使用Random庫以及使用Secrets庫來從列表中隨機挑選一個或多個元素的不同實現方法,並均給出了程式碼範例。

到此這篇關於如何利用Python隨機從list中挑選一個元素的文章就介紹到這了,更多相關 Python從list中挑選元素內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com