python实现伯努利分布

可以使用numpy库中的random.binomial函数来实现伯努利分布，代码如下：

import numpy as np

def bernoulli(p, size=None):
return np.random.binomial(1, p, size)

其中p为概率参数，size为生成的随机数的数量。