Breaking News: Grepper is joining You.com. Read the official announcement!

write pandas dataframe to postgresql table psycopg2

Trained Tuna answered on March 21, 2022 Popularity 8/10 Helpfulness 4/10

answer write pandas dataframe to postgresql table psycopg2

write pandas dataframe to postgresql table psycopg2

Comment

# Write pandas df into postgres table
import psycopg2
import numpy as np
import psycopg2.extras as extras
import pandas as pd
  
def execute_values(conn, df, table):  
    tuples = [tuple(x) for x in df.to_numpy()]  
    cols = ','.join(list(df.columns))
    # SQL query to execute
    query = "INSERT INTO %s(%s) VALUES %%s" % (table, cols)
    cursor = conn.cursor()
    try:
        extras.execute_values(cursor, query, tuples)
        conn.commit()
    except (Exception, psycopg2.DatabaseError) as error:
        print("Error: %s" % error)
        conn.rollback()
        cursor.close()
        return 1
    print("the dataframe is inserted")
    cursor.close()  
  
conn = psycopg2.connect(
    database="ENVIRONMENT_DATABASE", user='postgres', password='pass', host='127.0.0.1', port='5432'
)
 
df = pd.read_csv('fossilfuels.csv')  
execute_values(conn, df, 'fossil_fuels_c02')

Popularity 8/10 Helpfulness 4/10 Language sql

Source: www.geeksforgeeks.org

Tags: dataframe json pandas postgresql psycopg2 python sql

Link to this answer
Share Copy Link