python 二项分布_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()

实验图：