python中numpy.random.choice函数

numpy.random.choice函数

numpy.random.choice函数的主要作用：从给定的一维数组随机抽取生成新数组。

用法

import numpy as np

np.random.choice(a, size=None, replace=True, p=None)

参数

• a：给定的一维数组；
• size：输出的形状，可以是int 或 tuple of ints；
• replace：表示是否重复抽取，默认值True表示重复抽取；
• p：一维数组，与输入的a相对应，表示a中对应元素的抽取概率。

实例

实例1

生成一维数组：

a = np.array(range(5))
a

输出：

array([0, 1, 2, 3, 4])

使用numpy.random.choice函数从a中抽取10个数(允许重复)：

np.random.choice(a,size=10)

输出：

array([4, 2, 3, 1, 3, 2, 3, 2, 3, 3])

实例2

依旧使用实例1中的一维数组a，从中抽取4个数(允许重复)，生成2*2的数组：

np.random.choice(a,size=(2,2))

输出：

array([[2, 3],
       [0, 0]])