来自时间序列数据框的 matplotlib

Question

假设我有一个这样的数据框：from pandas import DataFrameexample = {'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 pd

df = 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.groupby

df.groupby('store_id').plot(y=['tot_employees', 'hrs_per_employee'])