np.random.choice() 是 NumPy 中的一個隨機抽樣函數(shù)，用于從給定的一維數(shù)組中隨機抽取指定數(shù)量或指定概率的元素。該函數(shù)可以用于構(gòu)建模擬實驗、生成隨機數(shù)據(jù)集、數(shù)據(jù)抽樣等應(yīng)用場景。

np.random.choice(a, size=None, replace=True, p=None) 的參數(shù)如下：

• a：一維數(shù)組或整數(shù)，表示需要進(jìn)行抽取的數(shù)據(jù)源，當(dāng)為整數(shù)時，相當(dāng)于 np.arange(n)。
• size：整數(shù)或元組，表示輸出數(shù)組的大小。
• replace：布爾值，表示是否允許重復(fù)抽樣，默認(rèn)為 True（允許重復(fù)抽樣）。
• p：一維數(shù)組，表示每個元素被抽到的概率。如果未指定，則默認(rèn)為均勻分布。

例如：

import numpy as np
a = np.array([1, 2, 3, 4, 5])
print(np.random.choice(a))   # 隨機抽取一個元素
# 輸出：1 或 2 或 3 或 4 或 5，具體結(jié)果根據(jù)隨機結(jié)果而定
print(np.random.choice(a, size=3))   # 隨機抽取三個元素
# 輸出：[2, 5, 1] 或 [3, 4, 1] 或 ...，具體結(jié)果根據(jù)隨機結(jié)果而定
print(np.random.choice(a, size=3, replace=False))   # 無放回地隨機抽取三個元素
# 輸出：[5, 3, 1] 或 [4, 2, 1] 或 ...，具體結(jié)果根據(jù)隨機結(jié)果而定
print(np.random.choice(a, size=3, p=[0.1, 0.1, 0.2, 0.3, 0.3]))   # 按照概率分布隨機抽取三個元素
# 輸出：[4, 5, 4] 或 [3, 3, 5] 或 ...，具體結(jié)果根據(jù)隨機結(jié)果而定

如果 a 是一個常數(shù)，np.random.choice() 函數(shù)則等價于從 [0, a) 的整數(shù)集合中隨機抽取元素。這是因為在 Python 中，range(a) 和 np.arange(a) 都可以表示 [0, a) 的整數(shù)集合，因此當(dāng) a 為整數(shù)時，np.random.choice(a) 等價于 np.random.choice(np.arange(a))。

例如：

import numpy as np
print(np.random.choice(5))   # 等價于 np.random.choice(np.arange(5))
# 輸出：0 或 1 或 2 或 3 或 4，具體結(jié)果根據(jù)隨機結(jié)果而定

需要注意的是，np.random.choice() 函數(shù)返回的是一個新的數(shù)組，不會改變原數(shù)組。如果需要對原數(shù)組進(jìn)行修改，可以使用 np.random.shuffle() 函數(shù)來打亂原數(shù)組的元素順序，然后再按照需要取出一部分元素。

到此這篇關(guān)于np.random.choice()函數(shù)的文章就介紹到這了,更多相關(guān)np.random.choice()函數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論