码迷,mamicode.com
首页 > 编程语言 > 详细

Python 模拟伯努利试验和二项分布

时间:2019-11-30 18:51:58      阅读:130      评论:0      收藏:0      [点我收藏+]

标签:python   函数   image   numpy   pre   info   color   bin   bsp   

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 模拟伯努利试验和二项分布

标签:python   函数   image   numpy   pre   info   color   bin   bsp   

原文地址:https://www.cnblogs.com/shanger/p/11963396.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!