函數參數:
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])