numpy抽样函数 np.random.choice用法详解

顾名思义，抽样函数，定义如下：

def choice(a, size=None, replace=True, p=None):

参数说明：

a :待抽样的样本（一维数组或整数）
size: 输出大小，默认返回单个元素
replace : 抽样后的元素是否可重复，默认是
p: 每个样本点被抽样的概率，默认均匀抽样

举例如下：

从[1,2,3,4,5]中随机抽三个元素，可重复，概率分别为[0.1,0.1,0.2,0.1,0.5]

>>> a=[1,2,3,4,5]
>>> p=[0.1,0.1,0.2,0.1,0.5]
>>> np.random.choice(a,3,True,p)
array([5, 2, 5])

元素不可重复（即第三个参数replace设为false）：

>>> np.random.choice(a,3,False,p)
array([2, 3, 5])

若输入a为整型，则表示从0到a-1中的整数样本进行抽样，如：

>>> np.random.choice(5,3,True)
array([4, 1, 3])
>>> np.random.choice(5,3,True)
array([0, 4, 2])

第二个参数size不设置，则只返回单个元素，如：

>>> np.random.choice(5)
2
>>> np.random.choice(5)
1