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])