python - 处理来自 SQLAlchemy Join 的 Pandas DataFrame 构造函数中的重复列

Question

我知道read_csv有mangle_dup_cols但是我如何在发出后从 sqlalchemy 中的 sql join 执行相同的操作:

pd.DataFrame(result.fetchall(), columns=result.keys())

使用 df.info() 时出现错误因为重复的列名。

Answer 1

您可以创建自己的辅助函数来修改列名称。下面的代码是我从 io.parsers._infer_columns 复制的:

def mangle_dupe_cols(columns):
    counts = {}
    for i, col in enumerate(columns):
        cur_count = counts.get(col, 0)
        if cur_count > 0:
            columns[i] = '%s.%d' % (col, cur_count)
        counts[col] = cur_count + 1
    return columns

pd.DataFrame(result.fetchall(), columns=mangle_dupe_cols(result.keys()))