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()
上一篇:DolphinDB Database丨使用Window Join快速估计个股交易成本


下一篇:服务器推送消息与 websocket