将滚动平均值与数据一起绘制

Question

我有一个看起来像这样的 DataFrame：

####delays:
            Worst case  Avg case
2014-10-27    2.861433  0.953108
2014-10-28    2.899174  0.981917
2014-10-29    3.080738  1.030154
2014-10-30    2.298898  0.711107
2014-10-31    2.856278  0.998959
2014-11-01    3.118587  1.147104
...

我想绘制这个 DataFrame 的数据，以及数据的滚动平均值。我希望数据本身应该是一条虚线，而滚动平均值应该是一条实线。最坏情况列应为红色，而平均情况列应为蓝色。

我试过以下代码：

import pandas as pd
import matplotlib.pyplot as plt

rolling = pd.rolling_mean(delays, 7)
delays.plot(x_compat=True, style='r--')
rolling.plot(style='r')
plt.title('Delays per day on entire network')
plt.xlabel('Date')
plt.ylabel('Minutes')
plt.show()

不幸的是，这给了我 2 个不同的情节。一个带有数据，一个带有滚动平均值。此外，最坏情况列和平均情况列都是红色的。

我怎样才能让它工作？

Answer 1

你需要告诉熊猫你想在哪里绘图。默认情况下，pandas 会创建一个新图形。

只需修改这两行：

delays.plot(x_compat=True, style='r--')
rolling.plot(style='r')

作者：

ax_delays = delays.plot(x_compat=True, style='--', color=["r","b"])
rolling.plot(color=["r","b"], ax=ax_delays, legend=0) 

现在在第二行告诉 pandas 在 ax_delays 上绘图，并且不再显示图例。

要为 2 条线获得 2 种不同的颜色，只需使用 color 参数传递尽可能多的颜色（见上文）。