我正在使用python中的matplotlib绘制直方图,并想绘制一条表示数据集平均值的线,以虚线的形式覆盖在直方图上(或者也许也可以使用其他颜色)。关于如何绘制直方图上的线的任何想法?
我正在使用plot()命令,但不确定如何绘制垂直线(即,应为y轴赋予什么值?
谢谢!
Answers:
您可以使用plot
或vlines
绘制一条垂直线,但是从y轴的底部到顶部绘制一条垂直线,axvline
可能是最简单的函数。这是一个例子:
In [80]: import numpy as np
In [81]: import matplotlib.pyplot as plt
In [82]: np.random.seed(6789)
In [83]: x = np.random.gamma(4, 0.5, 1000)
In [84]: result = plt.hist(x, bins=20, color='c', edgecolor='k', alpha=0.65)
In [85]: plt.axvline(x.mean(), color='k', linestyle='dashed', linewidth=1)
Out[85]: <matplotlib.lines.Line2D at 0x119758828>
这是个老话题,只是次要的加法,但是我经常喜欢做的一件事就是在行旁边绘制平均值:
import numpy as np
import matplotlib.pyplot as plt
np.random.seed(6789)
x = np.random.gamma(4, 0.5, 1000)
result = plt.hist(x, bins=20, color='c', edgecolor='k', alpha=0.65)
plt.axvline(x.mean(), color='k', linestyle='dashed', linewidth=1)
min_ylim, max_ylim = plt.ylim()
plt.text(x.mean()*1.1, max_ylim*0.9, 'Mean: {:.2f}'.format(x.mean()))