Flask+Pandas导入CSV到PostgreSQL指定列报IndexError求助

嘿，我来帮你搞定这个问题！

你碰到的IndexError: string index out of range其实是因为遍历DataFrame的方式错了：当你写for row in df时，Python实际是在遍历DataFrame的列名（一个个字符串），而不是每行的数据。所以row是列名字符串，当你尝试row[5]时，如果这个列名的长度不足5个字符，自然就会触发索引越界错误。

我帮你调整了代码，不仅解决了遍历问题，还优化了SQL查询的安全性（用参数化查询避免SQL注入风险）：

bkfile = request.files['bk_file']
conn = psycopg2.connect("host=localhost dbname=saveory user=user password=pass")
cur = conn.cursor()
df = pd.read_csv(bkfile)

# 使用iterrows()遍历每行数据，返回(index, row_series)组合
for idx, row in df.iterrows():
    # 用pd.isna()判断空值，兼容NaN、None等多种空值情况
    if pd.isna(row[5]) or row[5] == '':
        # 参数化查询，把变量作为第二个参数传给execute，避免SQL注入
        query = """
            INSERT INTO main_table(column2, column4, column5, column6)
            VALUES (%s, %s, %s, 'charge')
        """
        cur.execute(query, (row[1], row[3], row[6]))
    if pd.isna(row[6]) or row[6] == '':
        query = """
            INSERT INTO main_table(column2, column4, column5, column6)
            VALUES (%s, %s, %s, 'credit')
        """
        cur.execute(query, (row[1], row[3], row[5]))

conn.commit()
# 记得关闭游标和连接，避免资源泄漏
cur.close()
conn.close()

• 用列名代替索引：如果你的CSV有表头，建议直接用row['对应列名']代替row[索引]，比如row['用户ID']，这样代码更易读，也不会因为CSV列顺序变化出错。
• 批量插入提效：如果CSV数据量很大，逐行插入效率很低，可以把数据整理成列表后用executemany()批量插入，减少数据库交互次数。
• 添加异常处理：可以加个try-except块处理数据库操作中的异常（比如连接失败、数据格式错误），避免程序直接崩溃。

内容的提问来源于stack exchange，提问作者Tal Alterzon