Python数据处理小问题

Question

如何用python代码把下图中的表格1汇总成表格2

表格1

我想要达到的结果

表格2

Answer 1

使用pandas中的pivot_table函数去解决：

import pandas as pd
df=pd.read_excel('t0302.xlsx')
dfr=pd.pivot_table(values='bmi',index='code',columns='claim_type',data=df,aggfunc=sum,fill_value=0)
dfr=dfr.rename(columns={k:f'bmi_{k}' for k in dfr.columns})
print(dfr)

运行结果：

claim_type  bmi_12  bmi_21  bmi_31
code
a               30      20      10
b               40      50      40
c              105       0      20