pandas groupby sum 后转表格
时间: 2024-06-01 16:03:23 浏览: 109
假设我们有以下的DataFrame:
```
import pandas as pd
df = pd.DataFrame({
'Name': ['Alice', 'Bob', 'Charlie', 'Alice', 'Bob', 'Charlie'],
'Item': ['Apple', 'Banana', 'Apple', 'Banana', 'Apple', 'Banana'],
'Quantity': [3, 5, 2, 4, 1, 6]
})
print(df)
```
输出:
```
Name Item Quantity
0 Alice Apple 3
1 Bob Banana 5
2 Charlie Apple 2
3 Alice Banana 4
4 Bob Apple 1
5 Charlie Banana 6
```
我们想要对Name和Item进行分组,并计算Quantity的总和,可以使用groupby和sum方法:
```
grouped = df.groupby(['Name', 'Item']).sum()
print(grouped)
```
输出:
```
Quantity
Name Item
Alice Apple 3
Banana 4
Bob Apple 1
Banana 5
Charlie Apple 2
Banana 6
```
现在,我们可以使用unstack方法将其转换为表格形式:
```
pivoted = grouped.unstack()
print(pivoted)
```
输出:
```
Quantity
Item Apple Banana
Name
Alice 3 4
Bob 1 5
Charlie 2 6
```
这样,我们就得到了按Name和Item分组,并计算Quantity总和的表格形式。
阅读全文