Pandas DataFrames with NaNs equality comparison

Question

In the context of unit testing some functions, I\'m trying to establish the equality of 2 DataFrames using python pandas:

ipdb> expect
                            


        

Answer 1

df.fillna(0) == df2.fillna(0)

You can use fillna(). Documenation here.

from pandas import DataFrame

# create a dataframe with NaNs
df = DataFrame([{'a': 1, 'b': 2}, {'a': 5, 'b': 10, 'c': 20}])
df2 = df

# comparison fails!
print df == df2

# all is well 
print df.fillna(0) == df2.fillna(0)