np.random.choice
时间: 2023-05-18 09:07:35 浏览: 69
python np.random.choice方法
5星 · 资源好评率100%
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%,依此类推。
阅读全文