python - 如果另一列中的相应值也为 NaN，则将一列中的所有值设置为 NaN

Question

目标是通过将一列中的所有值设置为 NaN 来维护两列之间的关系。

具有以下数据框:

df = pd.DataFrame({'a': [np.nan, 2, np.nan, 4],'b': [11, 12 , 13, 14]})

     a   b
0  NaN  11
1    2  12
2  NaN  13
3    4  14

维护从 a 列到 b 列的关系，其中更新所有 NaN 值导致:

     a    b
0  NaN  NaN
1    2   12
2  NaN  NaN
3    4   14

实现所需行为的一种方法是:

df.b.where(~df.a.isnull(), np.nan)

有没有其他方法可以维持这样的关系？

Answer 1

您可以在 NaN 行上使用 mask。

In [366]: df.mask(df.a.isnull())
Out[366]:
     a     b
0  NaN   NaN
1  2.0  12.0
2  NaN   NaN
3  4.0  14.0

对于跨列的任何 NaN 使用 df.mask(df.isnull().any(1))