gpt4 book ai didi

python - 如何忽略 pandas assert frame equal 的索引比较

转载 作者:太空狗 更新时间:2023-10-29 17:01:35 27 4
gpt4 key购买 nike

我尝试比较以下两个将“check_index_type”设置为 False 的数据框。根据文档,如果设置为 False,则不应“检查 Index 类、dtype 和 inferred_type 是否相同”。我误解了文档吗?如何比较忽略索引并为以下测试返回 True?

我知道我可以重置索引,但我不想这样做。

https://pandas.pydata.org/pandas-docs/stable/generated/pandas.testing.assert_frame_equal.html

from pandas.util.testing import assert_frame_equal
import pandas as pd
d1 = pd.DataFrame([[1,2], [10, 20]], index=[0,2])
d2 = pd.DataFrame([[1, 2], [10, 20]], index=[0, 1])
assert_frame_equal(d1, d2, check_index_type=False)


AssertionError: DataFrame.index are different
DataFrame.index values are different (50.0 %)
[left]: Int64Index([0, 2], dtype='int64')
[right]: Int64Index([0, 1], dtype='int64')

最佳答案

如果你真的不关心索引是否相等,你可以按如下方式删除索引:

assert_frame_equal(d1.reset_index(drop=True), d2.reset_index(drop=True))

关于python - 如何忽略 pandas assert frame equal 的索引比较,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51655623/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com