在数据框列中从零到零之前的所有条目

Question

我想将所有非零值转换为零，直到每个组的 python 数据框列中最后一个零出现。

group | value | Result
 a    | 1     |  0
 a    | 2     |  0
 a    | 0     |  0
 a    | 1     |  0
 a    | 0     |  0
 a    | 1     |  1
 a    | 2     |  2
 b    | 1     |  0
 b    | 0     |  0
 b    | 2     |  2

我可以想到实现此目的的一种方法是反转value 列并将其上方的元素乘以每个组的元素，但是我不知道如何在 python 数据框中执行此操作。 任何帮助表示赞赏

Answer 1

您可以通过比较0 的值、将Series.iloc 与GroupBy.cumsum 的值交换以及最后比较不等于0 的值来测试所有值是否最后0，此掩码传递给Series.mask ：

m = df['value'].eq(0).iloc[::-1].groupby(df['group']).cumsum().ne(0)
df['New'] = df['value'].mask(m, 0)

换回原始订单的类似解决方案：

df1 = df.iloc[::-1]
m = df1['value'].eq(0).groupby(df1['group']).cumsum().ne(0).iloc[::-1]
df['New'] = df['value'].mask(m, 0)

print (df)
  group  value  Result  New
0     a      1       0    0
1     a      2       0    0
2     a      0       0    0
3     a      1       0    0
4     a      0       0    0
5     a      1       1    1
6     a      2       2    2
7     b      1       0    0
8     b      0       0    0
9     b      2       2    2