python - 在每行中查找前 N 个非空值

Question

如果我有这样的 Pandas 数据框:

 NaN NaN NaN  0  5  7  2  2  3  7  8
 NaN NaN  0   1  2  3  5  8  8 NaN 4
 NaN  0   3   6  9 NaN 4  6  1  5  1
 NaN NaN  0   1  2  3  5  8  8 NaN 2
 NaN NaN NaN  0  5  7  2  2  3  7  8
 NaN NaN  0   1  2  3  5  8  8 NaN 4

我如何只保留每行中的前五个非空值并将其余值设置为 nan 以便我得到一个如下所示的数据帧:

 NaN NaN NaN  0  5  7  2   2  NaN NaN NaN
 NaN NaN  0   1  2  3  5  NaN NaN NaN NaN
 NaN  0   3   6  9 NaN 4  NaN NaN NaN NaN
 NaN NaN  0   1  2  3  5 NaN NaN NaN NaN
 NaN NaN NaN  0  5  7  2   2  NaN NaN Nan
 NaN NaN  0   1  2  3  5  NaN NaN NaN NaN

Answer 1

您可以使用:

df.mask(df.notna().cumsum(axis=1).gt(5))