python scipy.stats实现各种常见的统计分布

import numpy as np
from scipy import stats
import matplotlib.pyplot as plt
from pylab import mpl
mpl.rcParams['font.sans-serif'] = ['SimHei']#中文雅黑字体
plt.rcParams['axes.unicode_minus']=False #显示负号X=np.arange(0,2,1)#[0,1)
p=0.7#科比投篮命中率
pList=stats.bernoulli.pmf(X,p)#在离散分布中，请将pmf改为pdf
print(pList)
plt.plot(X,pList,marker='o',linestyle='None')
'''
vlines用于绘制竖直线（vertical lines）,
参数说明：vline(x坐标值，y坐标最小值，y坐标最大值)
我们传入的X是一个数组，是给数组中的每个x坐标值绘制直线，
数值线y坐标最小值是0，y坐标最大值是对应的pList中的值'''
plt.vlines(X,(0,0),pList)

plt.xlabel('随机变量：科比投篮1次')
plt.ylabel('概率')
plt.title('伯努利分布：p=%.2f'% p)
plt.show()

stats.bernoulli.pmf(X,p)

#第1步，定义随机变量：5次抛硬币，正面朝上的次数
n=10  #做某件事的次数，例如科比三分投篮次数
p=0.75 #三分投篮命中率
X=np.arange(0,n+1,1)

#第2步：求对应分布的概率：概率质量函数（PMF）
#它返回一个列表，列表中每个元素表示随机变量中对应的概率
pList=stats.binom.pmf(X,n,p) #在离散分布中，请将pdf改为pmf
print(pList)
#第3步;绘图
plt.plot(X,pList,marker='o',linestyle='None')

plt.vlines(X,0,pList)
#x轴文本
plt.xlabel('随机变量：科比三分投篮命中次数')
#y轴文本
plt.ylabel('二项分布概率')
#标题
plt.title('二项分布：n=%i,p=%.3f'%(n,p))
#显示图形
plt.show()

stats.binom.pmf(X,n,p)

#第k次做某事，才取得第1次成功
#这里我们想知道5次表白成功的概率
k=5
#做某件事成功的概率，这里假设每次表白成功概率都是60%
p=0.6
X=np.arange(1,k+1,1)
#第2步：#求对应分布的概率：概率质量函数（PMF）
#它返回一个列表，列表中每个元素表示随机变量中对应值的概率
#分别表示表白第1次成功的概率，表白第2次成功的概率，表白第三次成功的概率
pList=stats.geom.pmf(X,p)
print(pList)

plt.plot(X,pList,marker='o',linestyle='None')
plt.vlines(X,0,pList)
#x轴文本
plt.xlabel('随机变量：表白第k次才首次成功')
#y轴文本
plt.ylabel('概率')
#标题
plt.title('几何分布：p=%.2f' %p)
#显示图形
plt.show()

stats.geom.pmf(X,p)

λ=3 #平均值：机器每周发生3次报警
k=10 #次数，现在想知道每周发生4次事故的概率
#包含了发生0次，1次，2次，3....,10次事故
X=np.arange(0,k+1,1)

#第2步： #求对应分布的概率：概率质量函数(PMF)
#它返回一个列表，列表中每个元素表示随机变量中对应值的概率
#分别表示发生1次，2次，3次，4次事故的概率
pList=stats.poisson.pmf(X,λ)
print(pList)

#第3步：绘图
plt.plot(X,pList,marker='o',linestyle='None')

plt.vlines(X,0,pList)
#x轴文本
plt.xlabel('随机变量：机器每周发生k次事故')
#y轴文本
plt.ylabel('概率')
#标题
plt.title('泊松分布：平均值mu=%i'%λ)
#显示图片
plt.show()

stats.poisson.pmf(X,λ)

#第1步：定义随机变量：
mu=0 #平均值
sigma=1 #标准差
X=np.arange(-5,5,0.1)
#第2步：概率密度函数（PDF）
y=stats.norm.pdf(X,mu,sigma)#连续分布用pdf,离散分布用pmf
#第3步：绘图
plt.plot(X,y)
plt.xticks(np.arange(-5, 5, 1))
#x轴文本
plt.xlabel('随机变量：x')
#y轴文本
plt.ylabel('概率')
#标题
plt.title('正态分布：$mu$=%.1f,$sigma^2$=%.1f'%(mu,sigma))
#网格
plt.grid()
#显示图形
plt.show()

stats.norm.pdf(X,mu,sigma)