np.random.choice()参数介绍
时间: 2023-06-24 08:06:58 浏览: 96
python np.random.choice方法
5星 · 资源好评率100%
`np.random.choice()` 是 numpy 包中的函数,用于从给定的一维数组中随机抽取元素。下面是它的参数介绍:
- `a`:一维数组或整数,表示待抽取的数据。如果是一个整数,则视为 np.arange(a) 的结果。
- `size`:表示要抽取的元素个数或抽取结果的形状,可以是整数或元组类型。
- `replace`:表示是否可以重复抽取同一个元素,True 表示可以,False 表示不可以。
- `p`:表示抽取每个元素的概率,如果不指定,则默认每个元素被抽取的概率相等。
- `axis`:表示从哪个轴抽取元素,默认为 None,表示从展开的数组中抽取。
下面是一个例子,展示如何使用 `np.random.choice()` 函数:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
print(np.random.choice(a)) # 随机抽取一个元素
print(np.random.choice(a, size=3, replace=False)) # 随机抽取三个不重复元素
```
阅读全文