PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

Convertendo colunas de pandas para a lista do PostgreSQL?


Assumindo que você já se conecta ao PostgreSQL e já possui a tabela no PostgreSQL. Ou visite este link https://wiki.postgresql.org/wiki/Psycopg2_Tutorial
import psycopg2

try:
    conn = psycopg2.connect("host='localhost' dbname='template1' user='dbuser' password='dbpass'")
except:
    print "I am unable to connect to the database"

Primeiro, abra o arquivo .csv.
>>> import csv
>>> with open('names.csv') as csvfile:
...     reader = csv.DictReader(csvfile)
...     for row in reader:
...         print(row['first_name'], row['last_name'])
...

Esse é o exemplo de https://docs.python.org/2/library/csv. html Altere a linha de impressão com insert no PostgreSQL.
>>> import psycopg2    
>>> cur.execute("INSERT INTO test (num, data) VALUES (%s, %s)",
    ...      (100, "abc'def"))

Você pode alterar (100, "abc'def") com (variable1, variable2) Veja este link http://initd.org/psycopg/docs/usage.html Ou no código de exemplo completo:
>>> import csv
>>> import psycopg2
>>> with open('names.csv') as csvfile:
...     reader = csv.DictReader(csvfile)
...     for row in reader:
...         cur.execute("INSERT INTO test (num, data) VALUES (%s, %s)", (variable1, variable2))
...

Espero que isso ajude...