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

Question

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

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

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

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

Answer 1

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

假设您有一个 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 模块运行良好。请查看它，如果这正是您要找的，请告诉我。