股票量化交易自学记录（1）tushare pro获取数据

    之前总是想用爬虫的方式获取数据，还是太麻烦了。发现可以使用tushare这样的平台调用数据，效率提升很大。

     Tushare pro的接口丰富，只需要注册得到一个token即可使用。但能调取的数据与积分值有关。基本数据只要初始积分就可以了，但调用的频率有所限制。得到的数据类型是dataframe类型。

import tushare as ts
import numpy as np
import pandas as pd
import os
import time
#下载6月25日当天所有的股票数据 开盘价 收盘价等
def stockdownload(): 
    df = pro.daily(trade_date='20210625')
    #df=ts.get_today_all()
    outpath='d:/olive/gp.csv'
    df.to_csv(outpath,sep=',',index=False)
    print(df)
#得到得的数据没有中文名字只有证券代码，把名字补到里面去
def getname():
    file = "D:\olive\gp.csv"
    data = pd.read_csv(file)
    data['name'] = ''
    for i in range(data.shape[0]):
        try:
           name=pro.stock_basic(ts_code=data['ts_code'][i])
           data['name'][i]=name['name'][0]
           time.sleep(0.3)
           print(data['name'][i])
        except:
           continue


    outpath = 'd:/olive/szx.csv'
    data.to_csv(outpath, sep=',', index=False)
    os.remove('d:/olive/gp.csv')
    src = 'd:/olive/szx.csv'
    dst = 'd:/olive/gp.csv'
    os.rename(src, dst)

pro = ts.pro_api('你注册得到的token')
stockdownload()
getname()