np.random.choice
时间: 2023-10-23 16:24:57 浏览: 130
python np.random.choice方法
5星 · 资源好评率100%
np.random.choice 是 NumPy 库中的一个函数,用于从给定的一维数组(或列表)中随机选择一个或多个元素。它有以下语法:
```python
numpy.random.choice(a, size=None, replace=True, p=None)
```
其中:
- a:一维数组或列表,表示从中选择元素的候选项。
- size:整数或元组,表示需要选择的元素的数量。如果未指定,则默认为 1。
- replace:布尔值,表示是否可以重复选择同一元素。如果为 True,则可以重复选择,否则不可以。默认为 True。
- p:一维数组,表示每个元素被选择的概率。如果未指定,则默认为每个元素被选择的概率相等。
例如,以下代码从 1 到 10 中随机选择 5 个元素:
```python
import numpy as np
a = np.arange(1, 11)
b = np.random.choice(a, size=5, replace=False)
print(b)
```
输出:
```
[ 8 6 5 10 7]
```
注意,由于 replace=False,所以不会选择重复的元素。
阅读全文