python 伯努利分布代码

伯努利分布是一种离散型概率分布，它只有两种取值，通常为0和1。在Python中，我们可以使用numpy库来模拟伯努利试验和伯努利分布。下面是一个简单的示例代码：

import numpy as np

# 模拟伯努利试验
p = 0.5  # 事件发生的概率
n = 10   # 试验次数
result = np.random.binomial(n, p, size=1)
print(result)  # 输出0或1，表示事件是否发生

# 生成伯努利分布
size = 100  # 生成的样本数量
samples = np.random.binomial(n=1, p=0.5, size=size)
print(samples)  # 输出一个包含0和1的数组，表示每个样本的取值

在上面的代码中，我们使用了numpy库的`random.binomial()`函数来模拟伯努利试验和生成伯努利分布。其中，`binomial()`函数的第一个参数表示试验次数，第二个参数表示事件发生的概率，第三个参数表示生成的样本数量。在模拟伯努利试验时，我们只生成了一个样本，因此将`size`参数设为1。在生成伯努利分布时，我们生成了100个样本，因此将`size`参数设为100。