要使用assert_frame_equal函数来断言两个相同的pandas DataFrame,可以按照以下步骤进行:
import pandas as pd
from pandas.testing import assert_frame_equal
data1 = {'col1': [1, 2, 3], 'col2': [4, 5, 6]}
df1 = pd.DataFrame(data1)
data2 = {'col1': [1, 2, 3], 'col2': [4, 5, 6]}
df2 = pd.DataFrame(data2)
assert_frame_equal(df1, df2)
完整的代码示例如下:
import pandas as pd
from pandas.testing import assert_frame_equal
# 创建两个相同的DataFrame对象
data1 = {'col1': [1, 2, 3], 'col2': [4, 5, 6]}
df1 = pd.DataFrame(data1)
data2 = {'col1': [1, 2, 3], 'col2': [4, 5, 6]}
df2 = pd.DataFrame(data2)
# 使用assert_frame_equal函数进行断言
assert_frame_equal(df1, df2)
如果两个DataFrame相同,以上代码将不会抛出异常。如果两个DataFrame不同,assert_frame_equal函数将抛出一个AssertionError异常。