python - Pandas pivot_table 日期

Question

我有一个带有日期列的 pandas DataFrame。它不是索引。

我想使用每个位置的每月计数聚合在数据框上创建一个 pivot_table。

数据看起来像这样:

['INDEX']                 DATE LOCATION  COUNT0          2009-01-02 00:00:00      AAH      11          2009-01-03 00:00:00      ABH      12          2009-01-03 00:00:00      AAH      13          2009-01-03 00:00:00      ABH      14          2009-01-04 00:00:00      ACH      1

我用过:

pivot_table(cdiff, values='COUNT', rows=['DATE','LOCATION'], aggfunc=np.sum)

调整值。我需要一种将 cdiff.DATE 转换为月份而不是日期的方法。我希望最终得到类似的东西:数据如下所示:

    MONTH LOCATION  COUNTJanuary      AAH      2January      ABH      2January      ACH      1

我在 cdiff.DATE 上尝试了各种 strftime 方法，但没有成功。它想将 应用于字符串，而不是系列对象。

Answer 1

我建议:

months = cdiff.DATE.map(lambda x: x.month)
pivot_table(cdiff, values='COUNT', rows=[months, 'LOCATION'],
            aggfunc=np.sum)

要获取月份名称，请传递不同的函数或使用内置的 calendar.month_name。要以您想要的格式获取数据，您应该对结果调用 reset_index，或者您也可以这样做:

cdiff.groupby([months, 'LOCATION'], as_index=False).sum()