np.random.choice用法
时间: 2023-10-23 13:31:48 浏览: 55
np.random.choice是NumPy库中的一个函数,用于从给定的一维数组或整数范围中随机选择元素。它的常见用法如下:
1. 从一维数组中随机选择元素:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
random_choice = np.random.choice(arr)
print(random_choice)
```
输出:
```
3
```
2. 从一维数组中随机选择多个元素:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
random_choices = np.random.choice(arr, size=3, replace=False)
print(random_choices)
```
输出:
```
[4 2 5]
```
这里的`size`参数指定了选择的元素个数,`replace=False`表示不允许重复选择。
3. 从整数范围中随机选择元素:
```python
import numpy as np
random_choice = np.random.choice(10)
print(random_choice)
```
输出:
```
7
```
这里的参数可以是一个整数,表示从0到该整数(不包含该整数)的范围内随机选择一个元素。
除了上述用法外,np.random.choice还支持其他参数,如设置元素概率分布、指定随机种子等。你可以查阅NumPy文档来了解更多详细信息。