Pandas dataframe в таблицу PostgreSQL с использованием psycopg2 без SQLAlchemy? - PullRequest
Новая
       13

Pandas dataframe в таблицу PostgreSQL с использованием psycopg2 без SQLAlchemy?

0 голосов
/ 04 февраля 2019

Я хотел бы записать кадр данных Pandas в таблицу PostgreSQL без использования SQLAlchemy .

Имя таблицы должно соответствовать имени переменной pandas или заменить таблицу, если она уже существует.Типы данных также должны совпадать.

Я бы хотел избежать функции to_sql SQLAlchemy по нескольким причинам. 

import pandas as pd
from getpass import getpass
import psycopg2

your_pass = getpass(prompt='Password: ', stream=None)
conn_cred = {
    'host': your_host,
    'port': your_port,
    'dbname': your_dbname,
    'user': your_user,
    'password': your_pass
}
conn = psycopg2.connect(**conn_cred)
conn.autocommit = True

my_data = {'col1': [1, 2], 'col2': [3, 4]}

def store_dataframe_to_postgre(df, schema, active_conn):
    # df = pandas dataframe to store as a table
    # schema = schema for the table
    # active_conn = open connection to a PostgreSQL db
    # ...
    # Bonus: require explicit commit here, even though conn.autocommit = True


store_dataframe_to_postgre(my_data, 'my_schema', conn)

Это должно быть результатом в Postgre db: 

SELECT * FROM my_schema.my_data;

   col1  col2
     1     3
     2     4

1 Ответ

0 голосов
/ 16 июня 2019

вы можете попробовать, но этот код в вашем: 

 cursor = conn.cursor()  
 cur.copy_from(df, schema , null='', sep=',', columns=(my_data))

ссылочный код: скопировать фрейм данных в таблицу postgres со столбцом, имеющим значение по умолчанию

...