[英]Confirming equality of two pandas dataframes?
如何断言以下两个数据帧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)
好像有点乱。 有没有更好的方法来进行这种比较?
您可以使用assert_frame_equal
而不检查列的assert_frame_equal
。
# 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)
使用优雅的@Divakar 的想法 - 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
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.