python - iloc[] 按值列

Question

我想对列中的值使用 iloc。

df1 = pd.DataFrame({'col1': ['1' ,'1','1','2','2','2','2','2','3' ,'3','3'],
                'col2': ['A' ,'B','C','D','E','F','G','H','I' ,'J','K']})

我想在每个列值中选择索引 2 作为数据框，结果如下

col1 col2
   1    C
   2    F
   3    K

非常感谢

Answer 1

使用GroupBy.nth :

df2 = df1.groupby('col1', as_index=False).nth(2)

替代 GroupBy.cumcount :

df2 = df1[df1.groupby('col1').cumcount().eq(2)]

---

print (df2)
   col1 col2
2     1    C
5     2    F
10    3    K