Python调取数据库postgresql中的数据并批量导出为csv文件

我在数据库postgresql里面有一张表，是股票数据，里面有一段时间、所有股票的日交易数据，我想通过本地的python（我用的spyder）调用里面的数据，实现：每只股票的时间序列数据存成一个csv文件。

这里面主要注意的有两点：
1）写循环
2）格式化
有需求的小伙伴可以先想想怎么做，我把我的代码贴在下面给大家做个参考，只是一个框架，具体可以根据自己需要进行调整~

import numpy as np
import pandas as pd
import psycopg2
from sqlalchemy import create_engine

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

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

# 读取数据并批量导出为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))