python - Pandas 仅选择 idxmin+1(从第 n 个到最后一个)值

Question

如何在 pandas 中使用 idxmin 仅选择倒数第 n 个索引？

df = data.loc[data.groupby("key1")["key2"].idxmin()] 

当然只给我最后一个值。

具有以下形式的数据框:

key1       key2
 a           1
 a           2
 a           3
 b           4
 b           5
 c           6

期望的结果应该是:

key1       key2
 a           2
 b           5

Answer 1

你可以这样做:

df[df.groupby('key1')['key2'].transform(pd.DataFrame.rank) == 2]

或者

df[df.groupby('key1')['key2'].apply(lambda x: x.rank() == 2)]

输出:

  key1  key2
1    a     2
4    b     5