熊猫分组和并显示值

时间:2020-11-11 16:10:34

标签: pandas group-by sum export-to-csv

我需要按两列进行分组,并对第三列求和。我的数据如下:

site        industry        spent
Auto        Cars             1000
Auto        Fashion          200
Auto        Housing          100
Auto        Housing          300
Magazine    Cars             100
Magazine    Fashion          200
Magazine    Housing          300
Magazine    Housing          500

我的代码:

df.groupby(by=['site', 'industry'])['Revenue'].sum()

输出为:

                       spent
site        industry
Auto        Cars             1000
            Fashion          200
            Housing          400
Magazine    Cars             100
            Fashion          200
            Housing          800

当我将其转换为csv时,我只会花一栏。我想要的输出与原始数据的格式相同,只需要对收入进行求和,并且需要查看列中的所有值。

1 个答案:

答案 0 :(得分:1)

尝试使用as_index=False

df = df.groupby(by=['site', 'industry'], as_index=False).sum()
print(df)


       site industry  spent
0      Auto     Cars   1000
1      Auto  Fashion    200
2      Auto  Housing    400
3  Magazine     Cars    100
4  Magazine  Fashion    200
5  Magazine  Housing    800