PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Conversion des colonnes pandas en liste PostgreSQL ?

Supposons que vous soyez déjà connecté à PostgreSQL et que vous ayez déjà la table dans PostgreSQL. Ou visitez ce lien 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"

Commencez par ouvrir le fichier .csv.

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

C'est un exemple de https://docs.python.org/2/library/csv. html Modifiez la ligne d'impression avec insert dans PostgreSQL.

>>> import psycopg2    
>>> cur.execute("INSERT INTO test (num, data) VALUES (%s, %s)",
    ...      (100, "abc'def"))

Vous pouvez changer (100, "abc'def") avec (variable1, variable2) Voir ce lien http://initd.org/psycopg/docs/usage.html Ou en exemple de code complet :

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

J'espère que cela vous aidera...