numpy.random.choice函數詳解

函數參數:

numpy.random.choice(a, size=None, replace=True, p=None)

參數 解釋
a 一維數組或整數。如果是個數組,則新的樣本元素來自這個數組;如果是整數,隨機樣本來自np.range(a)
size 單數或元組。如果是單數,輸出該數量元素的一維數組;如果是元組(m, n, k),則生成m×n×k個樣本,排列方式與元組相同。
replace 輸出的數字能否重複,False表示不能重複。即False表示不放回取樣,True表示放回取樣。
p None或數組。如果爲None,每個數選擇的概率相同;如果爲數組,數組p的長度與a的長度相同,數組p中元素對應爲a中每個元素取到的概率

例1:

從[0,1,2,3,4]中等概率放回取樣三個數。

np.random.choice(5, 3)

輸出:

array([0, 3, 4])

例2

從[0,1,2,3,4]中按照每個數取到的概率[0.1, 0, 0.3, 0.6, 0]放回取樣三個數:

np.random.choice(5, 3, p=[0.1, 0, 0.3, 0.6, 0])

輸出:

array([3, 3, 0])
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章