在直方图中绘制平均线

Question

我正在 python 中使用 matplotlib 绘制直方图，并且想绘制一条代表数据集平均值的线，以虚线形式覆盖在直方图上（或者其他颜色也可以）。关于如何在直方图上绘制一条重叠的线有什么想法吗？

我正在使用plot()命令，但不确定如何绘制垂直线（即我应该为y轴指定什么值？

谢谢！

Answer 1

您可以使用

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>

结果：

Answer 2

这是老话题和小补充，但我经常喜欢的一件事是也在线旁边绘制平均值：

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()))

产生以下结果：

Answer 3

我会查看数据集中的最大值（即直方图箱值），将该值乘以大于 1 的数字（例如 1.5），然后使用它来定义 y 轴值。这样，无论直方图中的值如何，它都会显示在直方图上方。