绘图滚动意味着与数据一起

时间:2015-04-02 16:22:36

标签: python pandas matplotlib

我有一个看起来像这样的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()

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

我怎样才能让它发挥作用?

1 个答案:

答案 0 :(得分:7)

你需要对大熊猫说你想要的地方。默认情况下,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) 

在第2行,你现在告诉大熊猫在ax_delays上绘图,并且不再显示图例。

要为2行提供2种不同的颜色,只需使用color参数传递尽可能多的颜色(参见上文)。