根据另一个数据框中的布尔值设置一个数据框中的值

Question

我有一个可以用以下代码复制的 MWE：

import pandas as pd
a = pd.DataFrame([[1,2],[3,4]], columns=['A', 'B'])
b = pd.DataFrame([[True,False],[False,True]], columns=['A', 'B'])

这会创建以下数据框：

In [8]: a
Out[8]: 
   A  B
0  1  2
1  3  4

In [9]: b
Out[9]: 
       A      B
0   True  False
1  False   True

我的问题是，如何根据数据框 B 中的布尔值更改数据框 A 的值？

例如，如果我想在数据框 A 中创建 NAN 值，而在数据框 B 中有 False 的实例？

Answer 1

如果需要将False 替换为NaN:

print (a[b])
     A    B
0  1.0  NaN
1  NaN  4.0

或：

print (a.where(b))
     A    B
0  1.0  NaN
1  NaN  4.0

如果需要，将True 替换为NaN：

print (a[~b])
     A    B
0  NaN  2.0
1  3.0  NaN

或：

print (a.mask(b))
     A    B
0  NaN  2.0
1  3.0  NaN

您也可以使用带有一些标量值的where 或mask：

print (a.where(b, 7))
   A  B
0  1  7
1  7  4

print (a.mask(b, 7))
   A  B
0  7  2
1  3  7

print (a.where(b, 'TEST'))
      A     B
0     1  TEST
1  TEST     4