Psycopg2 连接 sql 数据库到 pandas 数据框

Question

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

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

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

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

Answer 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 模块工作正常。请看一下，如果这就是您要找的，请告诉我。

Answer 2

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

import pandas.io.sql as sqlio

df = sqlio.read_sql_query(query, connection)

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

Psycopg2 连接 sql 数据库到 pandas 数据框

Psycopg2 connection sql database to pandas dataframe

python

sql

psycopg2

python-3.x

pandas