python - 保留列的首选值并删除不太受欢迎的值

Question

数据框，df:

ID status  year
1    0     2000
1    1     2000
2    0     2001
3    1     2002
3    0     2002
4    1     2002 

当“1”状态可用于同一年的同一 ID 时，我想删除“0”状态，例如:

ID status  year
1    1     2000
2    0     2001
3    1     2002
4    1     2002 

我使用了以下代码:

df.drop_duplicates(subset=['year', 'id'], keep="first")

但是上面的代码删除了一些非零行。

Answer 1

请戴口罩

非重复年份
和

状态=1

下面的代码

df[(~df.duplicated(subset=['year'], keep=False))|(df['status'].eq(1))]



    ID  status  year
1   1       1  2000
2   2       0  2001
3   3       1  2002
5   4       1  2002