Psycopg2 将 sql 数据库连接到 pandas 数据框答案

【问题标题】：Psycopg2 connection sql database to pandas dataframePsycopg2 将 sql 数据库连接到 pandas 数据框
【发布时间】：2022-07-11 21:18:05
【问题描述】：

我正在做一个项目，我正在使用psycopg2 连接从数据库中获取数据，就像这样，

cursor = connection.execute("select * from table")
cursor.fetchall()

现在从表中获取数据后，我正在运行一些额外的操作来将数据从游标转换为 pandas 数据框。我正在寻找一些库或更强大的方法来将数据从 psycopg2 连接转换为 pandas 数据帧。

任何指导帮助将不胜感激。谢谢

【问题讨论】：

标签： python sql python-3.x pandas psycopg2

【解决方案1】：

您可以使用 pandas sqlio 模块在 pandas 数据框中运行和保存查询。

假设你有一个 psycopg2 connection 的连接，那么你可以像这样使用 pandas sqlio。

import pandas.io.sql as sqlio
data = sqlio.read_sql_query("SELECT * FROM table", connection)
# Now data is a pandas dataframe having the results of above query.
data.head()

对我来说，sqlio pandas 模块运行良好。请看一下，如果这是您要找的，请告诉我。

【讨论】：

我刚刚测试过，它按我想要的方式工作。非常感谢您的快速回复。
很高兴能为您提供帮助。请将答案标记为已接受。谢谢

【解决方案2】：

这可能对您的情况有所帮助：

import pandas.io.sql as sqlio

df = sqlio.read_sql_query(query, connection)

在你的情况下，query = "select * from table"

【讨论】：