pandas - 大熊猫分组并总和显示值

Question

我需要按两列分组并对第三列求和。我的数据如下所示：

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 时，我只得到一列 - 已花费。我想要的输出与原始数据的格式相同，仅需要对收入求和，并且我需要查看列中的所有值。

标签： pandasgroup-bysumexport-to-csv

Answer 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