python - 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

我需要按国家、次级和来源的记录分组数。

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')