Python计算狄利克雷分布(Dirichlet Distribution)密度估计

数学推导、狄利克雷分布和狄利克雷函数等可以参考：https://www.datalearner.com/blog/1051459673766843#5%E3%80%81%E5%90%8E%E9%AA%8CDirichlet%E8%BF%87%E7%A8%8B

示例代码

from scipy.stats import dirichlet
import numpy as np

weight_array = np.array([9.98104384, 0.36104064, 9.86069696])

dir_mean = dirichlet.mean(alpha=dirichlet.mean(alpha=weight_array.tolist()))
print(dir_mean)

得到结果：

[0.49404305 0.01787084 0.48808611]