确认两个大pandas数据框是否相等？

小编典典

确认两个大pandas数据框是否相等？

python

如何断言以下两个数据帧df1和df2是相等的？

import pandas as pd
df1 = pd.DataFrame([1, 2, 3])
df2 = pd.DataFrame([1.0, 2, 3])

的输出df1.equals(df2)是False。到目前为止，我知道两种方法：

print (df1 == df2).all()[0]

要么

df1 = df1.astype(float)
print df1.equals(df2)

似乎有点混乱。有没有更好的方法进行比较？

阅读 220

2020-12-20

共1个答案

小编典典

您可以使用assert_frame_equal而不检查列的dtype。

# Pre v. 0.20.3
# from pandas.util.testing import assert_frame_equal

from pandas.testing import assert_frame_equal

assert_frame_equal(df1, df2, check_dtype=False)

2020-12-20