np.random.choice()
时间: 2023-12-07 10:37:51 浏览: 138
`np.random.choice()`是一个用于从给定的一维数组中生成随机样本的函数。它可以从数组中随机选择元素,也可以根据指定的概率分布进行选择。下面是一些使用`np.random.choice()`的例子:
1.从数组中随机选择元素:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
random_element = np.random.choice(arr)
print(random_element) # 输出:随机选择的一个元素
```
2.从数组中随机选择多个元素:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
random_elements = np.random.choice(arr, size=3, replace=False)
print(random_elements) # 输出:随机选择的三个元素
```
3.根据指定的概率分布进行选择:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
probabilities = [0.1, 0.2, 0.3, 0.2, 0.2]
random_elements = np.random.choice(arr, size=3, replace=False, p=probabilities)
print(random_elements) # 输出:根据概率分布随机选择的三个元素
```
阅读全文