当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python NumPy choice方法用法及代码示例


NumPy 的 choice() 方法返回随机样本数组。

参数

1.a | int 或类似一维数组

  • 如果给出 int，则生成 0(含)和 int(不含)之间的随机整数。

  • 如果给出类似数组，则从类似数组中随机选择元素。

2. size | int 或 int 的元组 | optional

  • 如果给出int，则size 表示要生成的随机数的数量。

  • 如果给定元组，则 size 表示输出形状。

默认情况下，返回单个值。

3. replace | boolean | optional

  • 如果是 True ，则该方法将进行放回采样，即一个元素可以被多次选择。

  • 如果是 False ，则该方法将进行无放回采样，即一个元素只能被选择一次。

默认情况下，replace=True 。

4. p |数字的一维数组 | optional

将用于生成 a 的概率。默认情况下，将使用均匀分布。

返回值

如果提供size，则返回NumPy 数组。

例子

生成 0(含)和 5(不含)之间的随机整数

要生成 0(含)和 5(不含)之间的单个数字：

np.random.choice(5)



3

生成 0(含)和 5(不含)之间的多个随机整数

要生成 0(含)和 5(不含)之间的多个数字：

np.random.choice(5, size=3)



array([4, 4, 2])

通过替换生成多个随机整数

要通过替换生成多个数字：

np.random.choice(5, size=3)   # replace=True



array([2, 4, 2])

请注意值 2 是如何被选择两次的。

生成多个随机整数而不进行替换

要生成多个数字而不进行替换：

np.random.choice(5, size=3, replace=False)



array([4, 2, 1])

在这里，随机选择的值保证是唯一的。

从数组中随机选择值

从给定数组中随机选择两个值：

np.random.choice([2,4,6,8], size=2)



array([4, 2])

相关用法


注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | choice method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。