如何排序Pandas DataFrame索引以通过pd.testing.assert_frame_equal测试

问题场景 #

自定义函数涉及DataFrame分组、melt和join操作，用pd.testing.assert_frame_equal做单元测试时，np.array_equal已验证两个DataFrame的值完全一致，但断言始终失败。

错误1：索引类型不匹配 #

• 错误表现：左右DataFrame索引类型不一致（左为Int64Index，右为RangeIndex）
• 临时解决：通过right.index = list(right.index)转换右侧索引类型，消除类型差异

错误2：索引顺序不符 #

• 报错原因：函数内虽调用了sort_index，但操作的是DataFrame副本，未作用到最终返回的对象上
• 最终解决：在函数中确保对目标DataFrame调用sort_index（比如使用链式调用df = df.sort_index()，或直接在原对象上执行操作），使返回结果的索引顺序与预期一致

关键提醒 #

pd.testing.assert_frame_equal的校验逻辑比np.array_equal严格得多，不仅校验值，还会检查索引的类型、顺序，甚至列的顺序等元信息，测试时需注意这些细节。

内容的提问来源于stack exchange，提问作者David Siret Marqués