在python中整洁地绘制PMF

Question

是否有一个库可以帮助我巧妙地绘制python中样本的概率质量函数，如下所示：

Answer 1

通过matplotlib.pyplot的stem模块

matplotlib.pyplot.stem（* args，** kwargs）

from matplotlib.pyplot import stem

stem(y, linefmt='b-', markerfmt='bo', basefmt='r-')
stem(x, y, linefmt='b-', markerfmt='bo', basefmt='r-')

或更接近金属

#!/usr/bin/env python
from pylab import *

x = linspace(0.1, 2*pi, 10)
markerline, stemlines, baseline = stem(x, cos(x), '-.')
setp(markerline, 'markerfacecolor', 'b')
setp(baseline, 'color','r', 'linewidth', 2)

show()

这里