Skriv hurtig panda-dataramme til postgres

Din anden tilgang bør være meget hurtig.

Der er to problemer med din kode:

1. Efter at have skrevet csv'en til f du er placeret i slutningen af ​​filen. Du skal sætte din position tilbage til begyndelsen, før du begynder at læse.
2. Når du skriver en csv, skal du udelade overskriften og indekset

Sådan skal din endelige kode se ud:

import io
f = io.StringIO()
pd.DataFrame({'a':[1,2], 'b':[3,4]}).to_csv(f, index=False, header=False)  # removed header
f.seek(0)  # move position to beginning of file before reading
cursor = conn.cursor()
cursor.execute('create table bbbb (a int, b int);COMMIT; ')
cursor.copy_from(f, 'bbbb', columns=('a', 'b'), sep=',')
cursor.execute("select * from bbbb;")
a = cursor.fetchall()
print(a)
cursor.close()