python - 根据Python中的条件在列中添加值

Question

Source_No  Parent   Type    Amt
123         123     Tail    100
456         123     Coll    100
789         123     Coll    100

我希望代码创建一个新列一个新列“Tot_Amt”，它将添加所有 Source_No 的 Amt，其中 Parent = Source_No 和其他

预期输出。

Source No  Parent   Type    Amt     Tot_Amt
123         123     Parent   100     200
456         123     Child    100      0
789         123     Child    100      0

标签： pythonpandas

Answer 1

干得好：

df["Tot_Amt"] = (df.groupby(["Parent"])["Amt"].sum() - df.groupby(["Source_No"])["Amt"].sum()).fillna(0).reset_index(drop=True)