python - 根据 Python Pandas 中 DataFrame 中的值求和？

Question

您好，我有如下 DataFrame：

df = pd.DataFrame({"ID" : ["1", "2", "1", "1"],
                   "code" : ["T", "N", "N", "T"],
                   "amount" : [100, 200, 300 ,400]})

我需要计算

1. sumT = 每个“ID”的“金额”之和，其中“代码”==“T”
2. sumN = 每个“ID”的“金额”之和，其中“代码”==“N”

所以我需要如下结果：

标签： pythonpandasdataframesum

Answer 1

尝试这个：

new_df.groupby(['ID','code']).agg(sum).unstack()
new_df.columns = ['Sum T','Sum N']
new_df = new_df.reset_index()