Pandas DataFrameGroupBy 导出到 Excel

Question

我正在尝试将分组记录的计数导出到 Excel。

我有一个看起来像这样的 CSV 文件：

Date        Country   Sub  Source
2014-09-11    US      1    source1
2014-09-11    US      2    source2
2014-09-11    UK      1    source3
2014-09-11    US      1    source1
2014-09-11    IN      3    source4

我需要按 Country、Sub 和 Source 分组的记录数。

df = pd.read_csv('log.csv',sep='\t')
count = df.groupby(['Country','Sub','Source']).size()
count.to_excel('report.xls', index=False)

但是当我运行它时出现错误： AttributeError: 'Series' object has no attribute 'to_excel'

是否不允许将 DataFrameGroupBy 导出到 Excel？还有一种方法可以将带有计数信息的分组记录插入到 MySQL 数据库中吗？

Answer 1

df = pd.read_csv('log.csv',sep='\t')
df = pd.DataFrame(df.groupby(['Country','Sub','Source']).size(),columns=['Count'])

你会得到：

                     Count
Country Sub Source        
IN      3   source4      1
UK      1   source3      1
US      1   source1      2
        2   source2      1

然后通过以下方式保存到excel：

df.to_excel('output.xls')