python - 绘制滚动平均值和数据

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

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

我怎样才能让它工作？

Answer 1

你需要告诉 pandas 你想绘制的地方。默认情况下，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 参数传递尽可能多的颜色(见上文)。