numpy.random.choice(a, size=None, replace=True, p=None)
功能:随机选取一维数组中的几个数
- a:待选数组
- size:选择的个数
- replace:True表示可以取相同数字。(默认)
- False表示取不同数字。
- p:数组中的元素被取到的概率
例:
arr = ['dog', 'rabbit', 'pig', 'cat']
b=np.random.choice(arr, 5, p=[0.5, 0.1, 0.1, 0.3])
>>['dog' 'dog' 'dog' 'dog' 'pig']