从数据库中的表获取数据

Gabriel L'Heureux 发表于 Dev

加布里埃尔·勒·赫鲁

我想从postgresql数据库中提取数据并在脚本中使用该数据（以数据框格式）。这是我最初的尝试：

from pandas import DataFrame
import psycopg2

conn = psycopg2.connect(host=host_address, database=name_of_database, user=user_name, password=user_password)

cur = conn.cursor()

cur.execute("SELECT * FROM %s;" % name_of_table)

the_data = cur.fetchall()

colnames = [desc[0] for desc in cur.description]

the_frame = DataFrame(the_data)
the_frame.columns = colnames

cur.close()
conn.close()

注意：我知道我不应该使用“字符串参数插值（％）将变量传递给SQL查询字符串”，但这对我来说非常有用。

会有更直接的方法吗？

编辑：这是我从所选答案中使用的内容：

import pandas as pd
import sqlalchemy as sq

engine = sq.create_engine("postgresql+psycopg2://username:password@host:port/database")

the_frame = pd.read_sql_table(name_of_table, engine)

约翰·兹温克

熊猫可以直接从Postgres加载数据：

import psycopg2
import pandas.io.sql as pdsql

conn = psycopg2.connect(...)

the_frame = pdsql.read_frame("SELECT * FROM %s;" % name_of_table, conn)

如果您的熊猫最近（> = 0.14），则应将sqlalchemy引擎与read_sql_query/table（read_frame已弃用）一起使用：

import pandas as pd
import sqlalchemy
import psycopg2

engine = sqlalchemy.create_engine("postgresql+psycopg2://...")

the_frame = pd.read_sql_query("SELECT * FROM %s;" % name_of_table, engine)
the_frame = pd.read_sql_table(name_of_table, engine)

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。