从SQLAlchemy Join处理Pandas DataFrame构造函数中的重复列

发布于 2021-01-29 16:47:57

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

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

df.info()由于使用了重复的col名称,使用时出现错误。

关注者
0
被浏览
48
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    您可以创建自己的帮助程序功能来处理列名。我从以下代码中复制了以下代码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()))
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看