Wie kann man Numpy-Vergleiche mit NaN durchführen, um NaN anstelle von False zurückzugeben? ⇐ Python

[Guest]

Ich führe Vergleiche (Gleichheit) einiger Reihen durch, die einige NaN-Elemente und numerische Elemente enthalten. Ich möchte, dass jeder Vergleich mit einem NaN NaN anstelle von False zurückgibt – was ist die beste Numpy-Funktion, um dies zu tun?

Code: Select all

df = pd.DataFrame({'a': [np.NaN, np.NaN, 1], 'b': [np.NaN, 1, 1]})

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

gibt

Code: Select all

0    False
1    False
2     True
dtype: bool

wann ich es zurückgeben möchte

Code: Select all

0    NaN
1    NaN
2    1
dtype: float

oder

Code: Select all

0    NaN
1    NaN
2    True
dtype: bool