python量化投资（二）——从tushare读取交易日信息（trade_cal）数据

1. 具体注册tushre，获取token不多在这里介绍了，官网对操作及接口描述很清晰
2. tushare官网地址tushare
3. 从tushare返回的数据是pandas.DataFrame数据类型，为了能够将数据写入数据库，需要使用numpy将其转换成列表，逐行写入
4. 使用numpy中的array方法将返回的数据表转换成多维数组，再使用tolist方法将多维数组转换成列表
5. 本次示例获取2020年一年的数据进行分析
6. 代码如下：

import tushare as ts
import pymysql
import numpy as np

ts.set_token('你的token')

pro = ts.pro_api()
# 调用trade_cal接口，设置起始日期和终止日期
data = pro.query('trade_cal', start_date='20200101', end_date='20201231', fields='cal_date,is_open,pretrade_date')
print(data)

# 转换成列表
datalist = np.array(data).tolist()

# 打开数据库连接
db = pymysql.connect(host='localhost', port=3306, user='xxx', passwd='xxx', db='tushare', charset='utf8mb4')

cursor = db.cursor()
# SQL 插入语句
sql = "INSERT INTO trade_cal (cal_date,is_open,pretrade_date) VALUES (%s,%s,%s)"

try:
    # 执行sql语句
    cursor.executemany(sql, datalist)
    # 提交到数据库执行
    db.commit()
except Exception as e:
    print(e)
    # 如果发生错误则回滚
    db.rollback()
# 关闭游标
cursor.close()
# 关闭数据库连接
db.close()