python - 进行 Pandas 比较时，如何从 np.nan>np.nan 返回 np.nan？

Question

我有以下两个数据框:

a = pd.DataFrame([[1,2, 3],[4,3,6], [np.nan, 2, np.nan]])
     0  1    2
0  1.0  2  3.0
1  4.0  3  6.0
2  NaN  2  NaN

和

b = pd.DataFrame([[0,1,3],[5,3,5 ],[np.nan, np.nan, np.nan]])
     0    1    2
0  0.0  1.0  3.0
1  5.0  3.0  5.0
2  NaN  NaN  NaN

a>b 的比较结果:

       0      1      2
0   True   True  False
1  False  False   True
2  False  False  False

但是我希望输出看起来像这样:

     0      1      2
0   True   True  False
1  False  False   True
2   nan    nan     nan

2>np.nan 和 np.nan>np.nan 的比较结果应该都是 np.nan。 (或任何其他不同于 True 和 False 的随机值)

任何事情都会有帮助!

Answer 1

我们需要添加一个掩码

yourdf=a.gt(b).mask(a.isna()|b.isna(),'nan')
Out[153]: 
       0      1      2
0   True   True  False
1  False  False   True
2    nan    nan    nan