我有一个数据块,目前是一个n元组列表,但格式非常灵活,我想将其附加到Postgres表中-在本例中,每个n元组对应于DB中的一行。
到目前为止,我一直在做的是将所有这些内容写入CSV文件,然后使用postgres的COPY将所有这些内容批量加载到数据库中。这是可行的,但并不是最优的,我更希望能够直接从python中完成这一切。有没有一种方法可以在python中复制Postgres中的复制类型批量加载?
发布于 2009-12-09 05:34:56
如果您使用的是psycopg2驱动程序,游标提供了一个copy_to
和copy_from
函数,可以从任何类似文件的对象(包括StringIO
缓冲区)中读取数据。
psycopg2 source distribution附带的文件examples/copy_from.py和examples/copy_to.py中有一些示例。
以下摘录来自copy_from.py
示例:
conn = psycopg2.connect(DSN)
curs = conn.cursor()
curs.execute("CREATE TABLE test_copy (fld1 text, fld2 text, fld3 int4)")
# anything can be used as a file if it has .read() and .readline() methods
data = StringIO.StringIO()
data.write('\n'.join(['Tom\tJenkins\t37',
'Madonna\t\N\t45',
'Federico\tDi Gregorio\t\N']))
data.seek(0)
curs.copy_from(data, 'test_copy')
https://stackoverflow.com/questions/1869973
复制相似问题