如何按组绘制时间序列 DataFrame，并根据条件更改线型？

Question

我想知道如何在 DataFrame 中绘制时间序列，在同一轴上，每个组都有不同的线。我还想根据一些（布尔）条件更改线型。下面是一个例子：

import pandas as pd
import numpy as np
from datetime import datetime, timedelta
from dateutil.parser import parse
from matplotlib import pyplot as plt

df = pd.DataFrame({'value': np.random.rand(18), 
              'group': ['A']*9 + ['B']*9,
              'future': [0, 0, 0, 0, 0, 0, 1, 1, 1]*2}, 
         index=[parse('2018-1-5') + timedelta(days=i) for i in range(9)]*2)

结果数据帧：

            future group     value
2018-01-05       0     A  0.076445
2018-01-06       0     A  0.800821
2018-01-07       0     A  0.410351
2018-01-08       0     A  0.039647
2018-01-09       0     A  0.664102
2018-01-10       0     A  0.192097
2018-01-11       1     A  0.456182
2018-01-12       1     A  0.163256
2018-01-13       1     A  0.832768
2018-01-05       0     B  0.139014
2018-01-06       0     B  0.265024
2018-01-07       0     B  0.832062
2018-01-08       0     B  0.738957
2018-01-09       0     B  0.334888
2018-01-10       0     B  0.945192
2018-01-11       1     B  0.707845
2018-01-12       1     B  0.135183
2018-01-13       1     B  0.140647

按组绘图很容易：

df.groupby('group')['value'].plot(legend='True')
plt.show()

但是，当对应的值为future1时，我希望线条变为虚线。

这是我对解决方案的尝试：

present_data = df.loc[df['future'] == 0]
future_data = df.loc[df['future'] == 1]

present_data.groupby('group')['value'].plot(legend='True')
future_data.groupby('group')['value'].plot(style='--')
plt.show()

但这不好。我想它可以通过手动设置颜色来修复（认为仍然会在行中留下间隙的问题），但肯定有比将 DataFrame 一分为二更好的方法？

非常感谢！

编辑：Will 的解决方案对我很有用，但是我仍然想知道如何正确创建图例。我希望为“A”和“B”组显示两条实线。

我试过这个：

legends = []
for idx,grp in df_plot.groupby(['group','future']):
    grp['value'].plot(style=s[grp['future'][0]],color=c[grp['group'][0]])
    if grp['future'][0] == 0:
        legends.append(grp['group'][0])
    else:
        legends.append('')
plt.legend(legends)

但这会导致没有添加标签的虚线：

附加None到图例列表也不起作用。如果我完全跳过附加，“A”和“B”不对应于正确的行：

legends = []
for idx,grp in df_plot.groupby(['group','future']):
    grp['value'].plot(style=s[grp['future'][0]],color=c[grp['group'][0]])
    if grp['future'][0] == 0:
        legends.append(grp['group'][0])
plt.legend(legends)
plt.show()

Answer 1

通过将分组合并为一个步骤，然后按组进行绘图，您可以更轻松地手动指定颜色和样式。为了消除差距，我认为您需要绘制一个额外的数据点。

extra=df[df.future==1]
extra = extra[extra.index == min(extra.index)] 
extra['future'] = 0
df_plot = pd.concat([df,extra])


s=['','--']
c={'A': 'red', 'B': 'blue'};

for idx,grp in df_plot.groupby(['group','future']):
    grp['value'].plot(style=s[grp['future'][0]],color=c[grp['group'][0]])