Question

我正在尝试检查一堆Pandas数据帧的索引是否相同。我编写了以下函数，它将数据帧的元组作为输入。

def chk_index_match(*dfs):
    ran_once = False
    for df in dfs:
        if not ran_once:
            ref_df = df
            ran_once = True
        else:
            if not(ref_df.index.equals(df.index)):
                return False
    return True

是否有内置的Pandas功能可以做类似的事情？或者更好的方法呢？

Answer 1

这是我发现的另一种方式，不确定它是否更好，但它使用更多现成的函数和更少的代码行：

import numpy as np
def chk_index_match(*dfs):
    arrays = [np.array(df.index) for df in dfs]
    return np.all([np.array_equal(arrays[0], a) for a in arrays])

检查多个pandas数据帧的索引是否相等

1 个答案: