如何从 numpy 概率数组中随机抽样？

Question

我有一个像这样的 numpy 数组：

arr = [.2 , .1, .2, .3, .2 ]

现在我想做的是从这个数组中采样，这样我得到某个索引的概率取决于索引处的概率。 例如，我得到索引 3 的概率是.3

任何人都知道任何漂亮的方法来做到这一点？

Answer 1

您可以使用np.random.choice通过将采样分布作为参数p来使用离散概率分布进行加权采样：

import numpy as np

x = [.2 , .1, .2, .3, .2 ]

# sample from `x` 100 times according to `x`
n_samples = 100
samples = np.random.choice(x, n_samples, p=x)