np.random.choice()
时间: 2023-10-23 07:25:29 浏览: 90
python np.random.choice方法
5星 · 资源好评率100%
`np.random.choice()`是NumPy中的一个函数,用于从给定的一维数组中随机选择元素。它的语法如下:
```python
np.random.choice(a, size=None, replace=True, p=None)
```
其中,参数:
- `a`: 一维数组,表示需要从中选择元素的数组。
- `size`: 表示需要选择的元素个数,可以是整数或者元组类型。
- `replace`: 表示是否可以重复选择同一个元素,取值为`True`或`False`。
- `p`: 表示每个元素被选择的概率,需要与`a`数组长度相同,如果不指定,则默认每个元素被选择的概率相等。
例如,以下代码将从一维数组`[1, 2, 3, 4, 5]`中随机选择3个元素:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
result = np.random.choice(a, size=3)
print(result)
```
输出:
```
[1 3 2]
```
注意,由于`replace`参数默认值为`True`,因此可能会选择重复的元素。如果不希望选择重复的元素,可以将`replace`设置为`False`。
阅读全文