python - 减去分组数据框中的行

Question

我有一个 pandas 数据框，如果它们具有相同的组，我需要减去行。

输入数据框:

<表类="s-表"><头>AB值<正文>A1B110.0A1B15.0A1B25.0A2B13.0A2B15.0A2B21.0

预期数据框:

<表类="s-表"><头>AB值<正文>A1B15.0A1B25.0A2B1-2.0A2B21.0

逻辑:例如，数据框的第一行和第二行在组 A1 和 B1 中，因此值必须为 10.0 - 5.0 = 5.0。 4º 和 5º 行也具有相同的组，因此该值必须为 3.0 - 5.0 = -2.0。

只减去具有相同 A 和 B 值的行。

谢谢!

Answer 1

尝试:

subtract = lambda x: x.iloc[0] - (x.iloc[1] if len(x) == 2 else 0)
out = df.groupby(['A', 'B'])['Value'].apply(subtract).reset_index()
print(out)

# Output:
    A   B  Value
0  A1  B1    5.0
1  A1  B2    5.0
2  A2  B1   -2.0
3  A2  B2    1.0