当前位置:首页 » python教程 » 正文

Python模拟伯努利试验和二项分布代码实例

看: 1335次  时间:2020-07-03  分类 : python教程

1、模拟 27 次投掷硬币的伯努利试验

代码:

from scipy import stats
import numpy as np
p = 0.5
# 生成冻结分布函数
bernoulliDist = stats.bernoulli(p) 

# 模拟 27 次伯努利实验
trails = bernoulliDist.rvs(27) 

# 查看结果
trails

2、模拟二项分布

代码

import numpy as np
from scipy import stats
import matplotlib.pyplot as plt

Ps = [0.5, 0.6, 0.7]
Ns = [20, 20, 20]
colors = ['blue', 'green', 'red']

# 模拟试验绘制图形
for p,n, c in zip(Ps, Ns, colors):
  binomDist = stats.binom(n, p)
  P_k = binomDist.pmf(np.arange(n + 1))

  label='p={},n={}'.format(p, n)
  plt.plot(P_k, '--',marker='o', label=label, ms=5)

plt.xlabel('X')
plt.ylabel('P(X)')
plt.legend()

plt.show()

结果

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持python博客。

标签:numpy  matplotlib  

<< 上一篇 下一篇 >>

搜索

推荐资源

  Powered By python教程网   鲁ICP备18013710号
python博客 - 小白学python最友好的网站!