如何用numpy/matplotlib绘制概率分布?
我想画一些概率函数,比如二项分布,但我找不到一个可以根据给定参数返回概率的函数。如果我自己写的话,我需要用到二项系数(我可以自己写这个),但我也没找到相关的函数。有没有什么简单或者快捷的方法可以做到这一点?
为了更清楚一点:我不想画直方图,也不想给直方图拟合一条线。
1 个回答
13
scipy.stats.binom.pmf
是用来计算二项分布的概率质量函数的。简单来说,它可以告诉你在一定次数的试验中,某个特定结果发生的概率。你可以为一系列的试验次数计算这个概率,并把结果画成图。比如说,如果你进行10次试验,成功的概率是0.1,你可以这样做:
import scipy, scipy.stats
x = scipy.linspace(0,10,11)
pmf = scipy.stats.binom.pmf(x,10,0.1)
import pylab
pylab.plot(x,pmf)