np.random.choice
时间: 2023-10-23 20:24:24 浏览: 112
np.random.choice 是一个 NumPy 函数,它可以从一个序列中随机地选择一个元素。你可以指定要从中选择的元素个数,并且可以指定一个概率分布来决定每个元素被选中的概率。
例如,假设你有一个列表 `items = [1, 2, 3, 4, 5]`,你可以使用以下代码随机地选择一个元素:
```
import numpy as np
selected_item = np.random.choice(items)
```
你也可以选择多个元素,例如:
```
selected_items = np.random.choice(items, size=3)
```
或者你可以指定一个概率分布,例如:
```
probs = [0.1, 0.2, 0.3, 0.4, 0.5]
selected_item = np.random.choice(items, p=probs)
```
在这种情况下,第一个元素被选中的概率是 10%,第二个元素被选中的概率是 20%,依此类推。
相关问题
np.random.choice用法
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文档来了解更多详细信息。
np.random.choice()
`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) # 输出:根据概率分布随机选择的三个元素
```
阅读全文