Python 计算给定值在给定高斯分布中的概率

背景

当我们用数据拟合一个符合高斯分布的概率密度函数之后,常常需要把另一个值带入该密度函数计算其概率。
我们使用Python可以轻松计算出这个数据。

参考资料

实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import numpy as np
import scipy.stats

#训练数据
data = [1, 2, 2, 3, 3, 3, 4, 4, 5]
#测试数据
test_value = 2
#平均值
mu = np.mean(data)
#标准差
std = np.std(data)
#概率
prob = scipy.stats.norm(mu, std).pdf(test_value)
print("prob.", prob)

#output: prob. 0.23745442481665455