如何根据没有循环的列（数据帧）替换另一列中组的所有相同值？

Question

我正在尝试将一组值中的所有相同值替换为0，如果与此组对应的另一列中有一个1。下面是我想要的输出示例，如果 Y 列中有 1，我希望输入列中由 1 组成的所有组都为 0。对于速度问题，我不想使用任何跨每个列值的循环（假设我的列可以有数千个值）。

Answer 1

用：

df['output']=(df['Y'].ne(1)        
                     .groupby(df['input'].ne(df['input'].shift()).cumsum())
                     .transform('all')
                     .mul(df['input']))

或

((~df['Y'].eq(1)
          .groupby(df['input'].ne(df['input'].shift()).cumsum())
          .transform('any')).mul(df['input']))