量化交易之Tushare Pro接口调用与数据库储存
股票数据存储
一.Tushare库安装
建议在anaconda中进行安装Tushare库
调用 pip install Tushare
安装完成后import tushare as ts 查看安装是否成功
ps:（遇到需要更新pip
则先python -m pip install --upgrade pip）
二.获取Tushare Pro的token
获取token点击：https://tushare.pro/register?reg=360055
注册既可获得token
三.数据获取与存储
数据存储
import pandas as pd
import tushare as ts
import pymysql
pymysql.install_as_MySQLdb()
def stock_daily_get(code_wm,start_dt,end_dt):
from sqlalchemy import create_engine
cn=create_engine(‘mysql+pymysql://root:密码@localhost:xxxx/stock?charset=utf8’)
ts.set_token(“your token”)
pro = ts.pro_api()
stock_fields = ‘ts_code,trade_date,open,high,low,close,pre_close,change,pct_chg,vol,amount’
df = pro.share_float(ts_code=code_wm, start_date=start_dt, end_date=end_dt,fields=stock_fields)
df.to_sql(‘stock_{}’.format(code_wm),cn,index=False)
print(’{}日线行情成功导入数据库’.format(code_wm))
接着打开数据库就可以查看刚刚存储的股票数据
结果展示
图片: 数据库初始密码
自己尝试调用一下吧
调用函数
if __name__=='__main__':
code_wm = '000001.SZ'
start_dt = '20190101'
end_dt = '20200319'
stock_daily_get(code_wm, start_dt, end_dt)