python二项分布代码

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

代码如下：

#-*- coding:utf-8 -*-  
import numpy as np
import matplotlib.pyplot as plt
import math
from scipy import stats

n = 20
p = 0.3
k = np.arange(0,41)
print k
print "*"*20
binomial = stats.binom.pmf(k,n,p)
print binomial


plt.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()

实验图：