python pivot 后面加sum

在使用Python中的pivot函数时，可以在其后面加上sum参数来计算每个组的总和。

例如，假设有一个名为df的数据框，它有三列：A、B和C。我们可以使用pivot函数将它们重塑成一个新的数据框，其中A是行，B是列，C是值。然后，我们可以使用sum参数来计算每个组的总和。

代码如下：

import pandas as pd

# 创建数据框
df = pd.DataFrame({'A': ['foo', 'foo', 'bar', 'bar', 'foo', 'foo'], 
                   'B': ['one', 'one', 'one', 'two', 'two', 'one'], 
                   'C': [1, 2, 3, 4, 5, 6]})

# 使用pivot函数重塑数据框
df_pivot = df.pivot(index='A', columns='B', values='C')

# 计算每个组的总和
df_sum = df_pivot.sum()

print(df_sum)

输出如下：

B
one    9
two    9
dtype: int64

可以看到，输出结果是一个Series，其中包含每个组的总和。在这个例子中，组one的总和是9，组two的总和也是9。