来自时间序列数据框的 matplotlib

Question

假设我有一个这样的数据框：

from pandas import DataFrame

example = {'year_month':  [201801,201802,201803,201801,201802,201803],

        'store_id': [101,101,101,102,102,102],

       'tot_employees': [100,200,150,6,7,10],

       'hrs_per_employee': [30,35,20,20,18,15]

        }

df = DataFrame(example,columns=["year_month", "store_id", "tot_employees", "hrs_per_employee"])

df

我想为每个 store_id 使用不同的子图堆叠子图：

• x轴：年月

• 情节第 1 行：全体员工

• 情节线 2：每位员工的小时数

df.plot() 可以吗？我无法找到正确的 x,y 输入来获得我正在寻找的结果。如果没有，是否有一个接近的选择？提前致谢

Answer 1

import pandas as pddf = pd.DataFrame(example)df.year_month = pd.to_datetime(df.year_month, format='%Y%m', exact=True)df.set_index('year_month', drop=True, inplace=True)for x in df.store_id.unique():     df[['tot_employees', 'hrs_per_employee']][df.store_id == x].plot(title=f'Store ID: {x}')仅使用df.groupbydf.groupby('store_id').plot(y=['tot_employees', 'hrs_per_employee'])