废话少说,上代码
# 以7次失败,3次成功的负二项分布可等价于3个几何分布为例
# 负二项分布
print(stats.nbinom.pmf(k=7,n=3,p=0.4))
# 几何分布
ret=0
for i in range(1,9,1):
ret_j=0
for j in range(1,10-i,1):
tem=stats.geom.pmf(k=j,p=0.4)*stats.geom.pmf(k=10-i-j,p=0.4)
ret_j+=tem
tem_i=ret_j*stats.geom.pmf(k=i,p=0.4)
ret+=tem_i
print(ret)
# results
0.06449725440000001
0.06449725439999998