python - pandas 提取每组第一行列值等于 1

Question

我有 df:

date   id   label   pred
1/1     1     0      0.2
2/1     1     1      0.5
1/1     2     1      0.9
2/1     2     1      0.3

我想要每个 id，当标签列等于 1 时获取第一行。例如 desire df:

date    id   label   pred
2/1      1    1       0.3
1/1      2     1      0.9

谢谢!

Answer 1

首先仅过滤具有 label=1 的行，然后通过 DataFrame.drop_duplicates 删除每个 id 的重复项 | :

df1 = df[df['label'].eq(1)].drop_duplicates('id')