python - pandas 适用于追加总行

Question

我已经经历了许多将总计行放在数据框末尾的示例。我只是想知道为什么下面的方法不能工作:

import pandas as pd

dict = {
    'a' : [1,2 ,3, 4, 5],
    'b' : [3, 5, 7, 9, 10]
}

df = pd.DataFrame(dict)
# print(df)

def func(x):
    x['sum'] = x.sum()

df2 = df.apply(lambda x: func(x), axis=0)

print(df2)

在此，x 始终是包含完整列的系列，并且我附加一个名为 sum 的索引。请指导。

编辑:如果我们可以使用 axis=1 计算总和，为什么我们不能使用 axis=0 来计算总和。

Answer 1

这里缺少函数中的return x:

def func(x):
    x['sum'] = x.sum()
    return x

df2 = df.apply(lambda x: func(x), axis=0)
print(df2)

      a   b
0     1   3
1     2   5
2     3   7
3     4   9
4     5  10
sum  15  34

但最简单的是使用setting with enlargement :

df.loc['sum'] = df.sum()