python - 尝试将数据插入 postgresql 时出现语法错误

Question

我正在尝试将已解析的 dta 数据插入到 postgresql 数据库中，每一行都是一个单独的变量表，它一直在工作，直到我在第二行中添加“recodeid_fk”。我现在尝试运行此代码时遇到的错误是:pg8000.errors.ProgrammingError: ('ERROR', '42601', '“imp”处或附近的语法错误')。

最终，我希望能够同时解析多个文件并将数据插入数据库，但如果有人能帮助我理解现在发生了什么，那就太棒了。我用的是Python 2.7.5，statareader来自pandas 0.12开发记录，我对Python的经验很少。

dr = statareader.read_stata('file.dta')
a = 2
t = 1
for t in range(1,10):
    z = str(t)
    for date, row in dr.iterrows():
        cur.execute("INSERT INTO tblv00{} (data, recodeid_fk) VALUES({}, {})".format(z, str(row[a]),29))
    a += 1
    t += 1
conn.commit()
cur.close()
conn.close()

Answer 1

针对您的特定错误...

语法错误可能来自字符串 {}，它们需要用引号引起来。 execute() 可以自动为您解决这个问题。替换

execute("INSERT INTO tblv00{} (data, recodeid_fk) VALUES({}, {})".format(z, str(row[a]),29))

execute("INSERT INTO tblv00{} (data, recodeid_fk) VALUES(%s, %s)".format(z), (row[a],29))

表名的完成方式与之前相同，但值将由 execute 填充，如果需要则插入引号。也许 execute 也可以填写表名，我们可以完全放弃 format，但这是一种不寻常的用法，我猜 execute 可能(错误地)在名字中间加上引号。

但还有更好的方法......

Pandas 包括 a function for writing DataFrames to SQL tables . Postgresql 尚不支持，但在简单的情况下，您应该能够假装您已连接到 sqlite 或 MySQL 数据库并且没有任何问题。

你在这里用 z 做什么？事实上，在继续下一个 for 循环之前，您将 z 从 '1' 循环到 '9'。循环应该嵌套吗？也就是说，您的意思是将内容 dr 插入到名为 tblv001 到 tblv009 的九个不同表中吗？

如果您的意思是将 dr 的不同部分放入不同的表中，请检查代码的缩进并加以说明。

无论哪种情况，上面的链接都应该负责 SQL 插入。

对编辑的回应

t、z 和a 似乎在做多余的事情。怎么样:

import pandas as pd
import string

...

# Loop through columns of dr, and count them as we go.
for i, col in enumerate(dr):
    table_name = 'tblv' + string.zfill(i, 3) # e.g., tblv001 or tblv010
    df1 = DataFrame(dr[col]).reset_index()
    df1.columns = ['data', 'recodeid_fk']
    pd.io.sql.write_frame(df1, table_name, conn)

我使用reset_index 将索引变成了一个列。 write_frame 不会保存新的(顺序)索引。