sql >> Base de Datos >  >> RDS >> PostgreSQL

Escriba el marco de datos de pandas rápidos en postgres

Su segundo enfoque debe ser muy rápido.

Hay dos problemas con su código:

  1. Después de escribir el csv en f usted está posicionado al final del archivo. Debe volver a colocar su posición al principio antes de comenzar a leer.
  2. Al escribir un csv, debe omitir el encabezado y el índice

Este es el aspecto que debería tener su código final:

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()