1. 程式人生 > 程式設計 >Python模擬伯努利試驗和二項分佈程式碼例項

Python模擬伯努利試驗和二項分佈程式碼例項

1、模擬 27 次投擲硬幣的伯努利試驗

程式碼:

from scipy import stats
import numpy as np
p = 0.5
# 生成凍結分佈函式
bernoulliDist = stats.bernoulli(p) 

# 模擬 27 次伯努利實驗
trails = bernoulliDist.rvs(27) 

# 檢視結果
trails

Python模擬伯努利試驗和二項分佈程式碼例項

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

結果

Python模擬伯努利試驗和二項分佈程式碼例項

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。