Python numpy 根据概率生成随机二进制值数组

Question

有什么方法可以在一行中执行以下操作吗？假设我有一系列概率，

p=np.array([0.75, 0.63, 0.33, 0.25, 0.47])

其中每个值对应选择1的概率，(1-p)对应选择0的概率，即0.75为选择1的概率，1 - 0.75为选择0的概率，0.63为选择1，选择 0 时为 1 - 0.63，依此类推。有没有更简单的方法可以让我完成以下操作，希望没有循环？

values = np.empty(p.shape)
for i, prob in enumerate(p):
    values[i] = np.random.binomial(1, prob)

我知道可以使用map来完成，但是有没有办法直接在numpy中完成它？

Answer 1

values = np.random.binomial(1, p)