Как сделать сравнение Numpy с использованием NaN, чтобы вернуть NaN вместо False? ⇐ Python

[Anonymous]

Я выполняю сравнения (равенство) некоторых рядов, которые содержат элементы NaN и числовые элементы. Я бы хотел, чтобы каждое сравнение, включающее NaN, возвращало NaN вместо False. Какая функция Numpy лучше всего подходит для этого?
df = pd.DataFrame({'a': [np.NaN, np.NaN, 1], 'b': [np.NaN, 1, 1]})

df['a'] == df['b']

дает
0 False
1 False
2 True
dtype: bool

когда я хочу, чтобы он вернулся
0 NaN
1 NaN
2 1
dtype: float

или
0 NaN
1 NaN
2 True
dtype: bool


Подробнее здесь: https://stackoverflow.com/questions/793 ... d-of-false