要使用assert_frame_equal
函数比较两个数据帧并在断言失败时显示差异,可以按照以下步骤进行操作:
import pandas as pd
from pandas.testing import assert_frame_equal
df1
和df2
:data1 = {'A': [1, 2, 3],
'B': [4, 5, 6]}
df1 = pd.DataFrame(data1)
data2 = {'A': [1, 2, 4],
'B': [4, 5, 7]}
df2 = pd.DataFrame(data2)
assert_frame_equal
函数比较两个数据帧并显示差异:try:
assert_frame_equal(df1, df2)
print("两个数据帧相等")
except AssertionError as e:
print("两个数据帧不相等:")
print(e)
在上述代码中,assert_frame_equal
函数用于比较df1
和df2
两个数据帧是否相等。如果两个数据帧不相等,将抛出一个AssertionError
异常,并且在except
块中打印出差异信息。