python二项分布代码

抛硬币，20次，每一次朝上的概率是0.3.连续几次正面朝上的概率图

代码如下：

#-*- coding:utf-8 -*-  import numpy as npimport matplotlib.pyplot as pltimport mathfrom scipy import statsn = 20p = 0.3k = np.arange(0,41)print kprint "*"*20binomial = stats.binom.pmf(k,n,p)print binomialplt.plot(k, binomial, 'o-')plt.title('binomial:n=%i,p=%.2f'%(n,p),fontsize=15)plt.xlabel('number of success')plt.ylabel('probalility of success', fontsize=15)plt.grid(True)plt.show()

实验图：