确认两个pandas数据帧的相等性?

时间:2016-07-05 21:03:25

标签: python pandas

如何声明以下两个数据帧df1df2相等?

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)

看起来有点凌乱。有没有更好的方法来进行这种比较?

2 个答案:

答案 0 :(得分:17)

您可以使用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)

答案 1 :(得分:3)

使用elegant @Divakar's idea - numpy' allclose()将成为数字的主要技巧:

In [128]: df1
Out[128]:
   0    s  n
0  1  aaa  1
1  2  aaa  2
2  3  aaa  3

In [129]: df2
Out[129]:
     0    s    n
0  1.0  aaa  1.0
1  2.0  aaa  2.0
2  3.0  aaa  3.0

In [130]: (np.allclose(df1.select_dtypes(exclude=[object]), df2.select_dtypes(exclude=[object]))
   .....:  &
   .....:  df1.select_dtypes(include=[object]).equals(df2.select_dtypes(include=[object]))
   .....: )
Out[130]: True

select_dtypes()将帮助您分隔字符串和所有其他数字dtypes