#导入包
#设置中文字体
from pylab import mpl
mpl.rcParams['font.sans-serif'] = ['FangSong'] # 指定默认字体
mpl.rcParams['axes.unicode_minus'] = False # 解决保存图像是负号'-'显示为方块的问题
#数组包
import numpy as np
#绘图包
import matplotlib.pyplot as plt
#统计计算包的统计模块
from scipy import stats#Step1：定义随机变量
X=np.arange(0,2,1)#Step2：定义概率分布
p = 0.5 #成功概率，即随机变量x=1时的概率值
pList = stats.bernoulli.pmf(X, p)
pList# Step3：绘制随机变量概率分布图
'''
marker：设置点的形状
linestyle：设置线条形状
'''
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, pList)
#x轴文本
plt.xlabel('随机变量：抛硬币1次')
#y轴文本
plt.ylabel('概率')
#标题
plt.title('伯努利分布：p=%.2f' %  p)
#显示图形
plt.show()

# step1：设置随机变量
n=5   #进行伯努利实验的次数
p=0.5 #伯努利实验中每次成功的概率
X=np.arange(0,n+1,1)#设置随机变量的值# step2：计算概率分布
pList=stats.binom.pmf(X,n,p)# step3：绘制随机变量概率分布图
plt.plot(X, pList, marker='o',linestyle='None') #设置线性
plt.vlines(X, 0, pList) #设置自变量、因变量的最小值和最大值
plt.xlabel('随机变量：抛硬币正面朝上次数')#x轴文本
plt.ylabel('概率')#y轴文本
plt.title('二项分布：n=%i,p=%.2f' % (n,p))#标题
plt.show()#显示图形

k = 5   #次数
p = 0.6 #每次成功的概率
#step1：随机变量
X = np.arange(1, k+1,1) #step2：计算每个事件发生的概率
pList = stats.geom.pmf(X,p)#step3：绘制随机变量概率分布图
#第3步，绘图
plt.plot(X, pList, marker='o',linestyle='None')
plt.vlines(X, 0, pList)
plt.xlabel('随机变量：表白第k次才首次成功')#x轴文本plt.ylabel('概率')#y轴文本
plt.title('几何分布：p=%.2f' % p)#标题
plt.show()#显示图形

#step1：随机变量设置
mu = 2   # 平均值：每天发生2次事故
k=4 #次数，现在想知道每天发生4次事故的概率
#包含了发生0次、1次、2次，3次，4次事故
X = np.arange(0, k+1,1)#step2：计算概率
pList = stats.poisson.pmf(X,mu)# step3：绘制概率分布图plt.plot(X, pList, marker='o',linestyle='None')
plt.vlines(X, 0, pList)
plt.xlabel('随机变量：某路口发生k次事故')
plt.ylabel('概率')
plt.title('泊松分布：平均值mu=%i' % mu)
plt.show()