Missing column names when importing data from database (python + postgre sql)

Question

I am trying to import some data from the database (Postgre SQL) to work with them in Python. I tried with the code below, which seems quite similar to the ones I've found on the internet.

import psycopg2
import sqlalchemy as db
import pandas as pd

engine = db.create_engine('database specifications')
connection = engine.connect()
metadata = db.MetaData()
data = db.Table(tabela, metadata, schema=shema, autoload=True, autoload_with=engine)
query = db.select([data])
ResultProxy = connection.execute(query)
ResultSet = ResultProxy.fetchall()
df = pd.DataFrame(ResultSet)

However, it returns data without column names. What did I forget?

Please include whatever imports are needed to make this work. — jjanes, May 10 '21 at 22:02
I've edited the code with the libraries needed to make this code work. — Vesnič, May 11 '21 at 06:04

score 0 · Accepted Answer · answered May 11 '21 at 11:35

0

It turned out the only thing needed is adding

columns = data.columns.keys()
df.columns = columns

There is a great debate about that in this thread.

answered May 11 '21 at 11:35

Vesnič

365
5
17

Missing column names when importing data from database (python + postgre sql)

1 Answers1