Python3：使用 Python 连接到 postgresql 数据库...如何让结果查询作为 pandas 数据框中的行返回？

Question

我已通过 Jupyter Notebook 将我的 python 连接到我的本地 postgresql 数据库。我能够成功运行 SELECT 查询并从我的表中提取数据。

但是，我想将我的 postgresql 表中的数据行显示为数据框，而不是我目前拥有的数据。

下面是我的代码：

conn = psycopg2.connect("dbname=juke user=postgres")

cur = conn.cursor()
cur.execute('SELECT * FROM albums')

for i in cur:
    print(i)

这是我的代码输出：

如何让输出显示为数据框中的行？

我从人们分享的推荐重复帖子中查看并尝试了一系列不同的可能解决方案。不幸的是，他们都没有工作。

Answer 1

使用熊猫创建数据框 - https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.read_sql_query.html:

import pandas as pd
conn = psycopg2.connect("dbname=juke user=postgres")
df = pd.read_sql_query("SELECT * FROM albums", conn)
df.head()