python - 计算组中元素之间的差异

Question

考虑 a 中每个类别恰好有两行的数据框:

d = pd.DataFrame({"a": ["a", "b", "c", "a", "b", "c"], "b": [1, 3, 1, 4, 2, 6]})
>   a   b
0   a   1
1   b   3
2   c   1
3   a   4
4   b   2
5   c   6

我想计算 b 与 a 之间的绝对差异:

    a   b
0   a   3
1   b   1
2   c   5

Answer 1

您可以将它们分组，然后计算差异和绝对值(链):

d.groupby('a').diff().abs().dropna()

索引可能不完全符合要求，但您可能可以弄清楚。