概率及常用概率分布的实现

二项分布

#-*- coding:utf-8 -*-
import numpy as np
from scipy import stats

n = input()     #n是样本数
p = input()     #p是抽样概率
n = int(n)
p = float(p)
########## Begin ##########
k = range(n+1)
binomial = stats.binom.pmf(k,n,p)
print (binomial)


########## End ###########

贝塔分布

from scipy.stats import beta
import numpy as np


a = input()     #贝塔分布的alpha值
b = input()     #贝塔分布的beta值
a = float(a)
b = float(b)

x=np.arange(0.1,1,0.1)     #给定的输入数据
########## Begin ##########
y=beta.pdf(x,a,b)
print(y)
########## End ##########

正态分布

import numpy as np

u = input()     # 均值μ
sig = input()     # 标准差δ
u = float(u)
sig = float(sig)

x = np.linspace(u - 3 * sig, u + 3 * sig, 50)     #给定的数据样本x

########## Begin ##########
y_sig = np.exp(-(x - u) ** 2 / (2 * sig ** 2)) / (np.sqrt(2 * np.pi) * sig)
print(y_sig)
########## End ##########