使用psycopg2将SQL中的数据直接作为数字而不是对象导入pandas数据帧。

Question

我可以像这样导入到数据帧中：

df=pd.DataFrame(cursor.fetchall(),columns=['name','ts','open','close']))

所有df列数据类型都将是object。

我可以将它们转换为正确的数字，如下所示：

df2=df.apply(pd.to_numeric, errors='ignore').info()

有没有办法在不做多次计算/代码行的情况下指定数据类型？

Answer 1

考虑使用SQLAlchemy引擎作为连接的pandas的read_sql：

import pandas as pd
import psycopg2
from sqlalchemy import create_engine

engine = create_engine("postgresql+psycopg2://user:password@/dbname")

df = pd.read_sql('SELECT * FROM myTable', con=engine)