本文详细介绍了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])
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
