给出python语言按概率抽样的代码
时间: 2023-02-08 20:15:51 浏览: 149
以下是在 Python 中按概率进行抽样的示例代码:
```
import random
# 定义概率分布
probabilities = [0.1, 0.2, 0.3, 0.4]
# 定义待抽样元素
elements = ['A', 'B', 'C', 'D']
# 使用 random.choices() 函数进行抽样
result = random.choices(elements, probabilities)
print(result)
```
请注意,在这个示例中,我们使用了 python3.6以上版本的random.choices()函数,它可以按照指定的概率分布进行抽样。
如果是python3.5以下的版本,可以使用numpy库中的函数numpy.random.choice(a, size=None, replace=True, p=None)来进行抽样,参数p就是概率分布。
阅读全文