从数据库调取数据的代码

从数据库调取数据的代码
#调取数据库
从数据库调取所需数据
打印数据。
1 import numpy as np
import pandas as pd
import psycopg2
from sqlalchemy import create_engine

1连接数据库：

conn = psycopg2.connect("dbname=数据库名字 user=postgres password=密码“）
curs = conn.cursor()

engine = create_engine(‘postgresql://postgres:密码@localhost:5432/数据库名字’)

#2 读取数据并批量导出为csv文件：

1、选择表中stockid中的非重复数据，即把代码生成一个list，为下面写循环做准备。

code = pd.read_sql_query(""" SELECT DISTINCT stockid FROM 表名
WHERE date=‘2018-04-17’;""", con=engine)
code_list = code[‘stockid’].tolist()

写循环、格式化输出

for i in code_list:
df = pd.read_sql_query(""“SELECT * FROM 表名
WHERE stockid=’%s’ AND year_month=‘2018-01’;”""%(i),con=engine)
df.to_csv(r’C:\Users\HP\Desktop\2018_01_%s_1min.csv’%(i))