python在df列中找到最近的非零元素

Question

我有 df:

id   number
1     5
1      0
1      0
1      2    
2      0
3      1

我想编写一个函数来填充 0 个值。我想为每个 id（对于每个组），当 number 列中的值为零时，搜索列中的壁橱非零值并返回该值。例如 id 1 用 2 填充第二行和第三行。如果我没有像 id 2 这样的值，就保持原样。 我怎样才能做到这一点？谢谢

Answer 1

您可以mask 0，然后使用 groupby.transform 获取每个组的最小值（如果您有负值，则需要相对最小值），然后使用它来填充na，最后为组填充 0而不是只有零：

s = df['number'].mask(df['number'].eq(0))

df['number2'] = s.fillna(s.groupby(df['id']).transform('min').fillna(0),
                         downcast='infer')

输出：

   id  number  number2
0   1       5        5
1   1       0        2
2   1       0        2
3   1       2        2
4   2       0        0
5   3       1        1