我有一个看起来像这样的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()
不幸的是,这给了我两个不同的情节。一个是数据,另一个是滚动平均值。此外,最坏情况列和平均情况列都是红色。
我怎样才能让它发挥作用?
答案 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
参数传递尽可能多的颜色(参见上文)。