np.random.choice用法

np.random.choice(a, size, replace, p) 其作用是按要求生成一個一維數組
a是生成一維數組的來源，可以是int類型，可以是數組，也可以是list
size 爲從a中抽取的個數，即生成數組的維度
replace 表示從a中是否不重複抽取，默認可重複
p 給出抽取概率，默認隨機
例子如下

>>> import numpy as np
>>> a1 = np.random.choice(7,5) # 從0~7中隨機選擇5個數組成一維數組
>>> a1
array([3, 1, 0, 1, 4])
>>> a2 = np.random.choice([0,1,2,3,4,5,6],5) # 從給定list中隨機選擇5個數組成一維數組
>>> a2
array([1, 6, 6, 6, 4])
>>> a3 = np.random.choice(np.array([0,1,2,3,4,5,6]),5) # 將list換成array數組依然可以運行，效果一致
>>> a3
array([3, 0, 5, 5, 3])
>>> a4 = np.random.choice([0,1,2,3,4,5,6],5,replace=False) # 上述均有重複，將replace設置爲False，即可按要求沒有重複的選取
>>> a4
array([0, 4, 3, 6, 5])
>>> a5 = np.random.choice(np.array([0,1,2,3,4,5,6]),5,p=[0.1,0.1,0.1,0.1,0.1,0.1,0.4]) 
# 給出選取概率p，注意p的維度和a的維度一致，並且p中概率和爲1
>>> a5
array([6, 3, 2, 6, 6])