python - Pandas 数据框 : Remove secondary upcoming same value

Question

我有一个数据框:

col1  col2
 a     0
 b     1
 c     1
 d     0
 c     1
 d     0

在 'col2' 上，我只想保留顶部的第一个 1 并将第一个下面的每个 1 替换为 0，输出为:

col1  col2
 a     0
 b     1
 c     0
 d     0
 c     0
 d     0

非常感谢。

Answer 1

你可以找到第一个1的索引，并将其他设置为0:

mask = df['col2'].eq(1)
df.loc[mask & (df.index != mask.idxmax()), 'col2'] = 0

要获得更好的性能，请参阅 Efficiently return the index of the first value satisfying condition in array .