查找两个数量之间差异最小的行对

Question

我有一个包含 a 和 t 列的数据框 df，其中“a”列包含字符串，“t”列包含整数。我想从数据框中选择所有行对，其中该行对的“a”列中的值相同，并且该对的“t”列中的值差异具有最小值。例如：

df = a    t
    abc   4
    abc   3
    def   2
    abc   1

我想得到以下结果：

df = a    t
     abc  4
     abc  3

我知道我们可以在同一个数据框中使用两个 for 循环，但我正在寻找更有效的解决方案。 感谢期待

Answer 1

你可以使用：

df = df.sort_values(['a', 't'], ascending=False)
diff_ = df['t']-df['t'].shift(-1)
min_idx = diff_[df['a'] == df['a'].shift(-1)].idxmin()
df.loc[min_idx:min_idx+1]

输出：

    a   t
0   abc 4
1   abc 3