Python模擬伯努利試驗和二項分佈程式碼例項
阿新 • • 發佈:2020-05-28
1、模擬 27 次投擲硬幣的伯努利試驗
程式碼:
from scipy import stats import numpy as np p = 0.5 # 生成凍結分佈函式 bernoulliDist = stats.bernoulli(p) # 模擬 27 次伯努利實驗 trails = bernoulliDist.rvs(27) # 檢視結果 trails
2、模擬二項分佈
程式碼
import numpy as np from scipy import stats import matplotlib.pyplot as plt Ps = [0.5,0.6,0.7] Ns = [20,20,20] colors = ['blue','green','red'] # 模擬試驗繪製圖形 for p,n,c in zip(Ps,Ns,colors): binomDist = stats.binom(n,p) P_k = binomDist.pmf(np.arange(n + 1)) label='p={},n={}'.format(p,n) plt.plot(P_k,'--',marker='o',label=label,ms=5) plt.xlabel('X') plt.ylabel('P(X)') plt.legend() plt.show()
結果
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。