load_data.py 文件源码

def download_from_tushare(code):
    '''
    #??????,????1:?????
    month :????
    m2 :???????????M2?(??)
    m2_yoy:???????????M2?????(%)
    m1:??(????M1)(??)
    m1_yoy:??(????M1)????(%)
    m0:?????(M0)(??)
    m0_yoy:?????(M0)????(%)
    cd:????(??)
    cd_yoy:????????(%)
    qm:???(??)
    qm_yoy:???????(%)
    ftd:????(??)
    ftd_yoy:????????(%)
    sd:????(??)
    sd_yoy:????????(%)
    rests:????(??)
    rests_yoy:????????(%)
    '''
    import tushare as ts
    path = './data/'

    # 1 day line
    hist_data = ts.get_hist_data(str(code))
    if hist_data is not None:
        hist_data.to_csv(path+"stock_data/"+str(code)+'.csv')
        return True
    else:
        return False
    # 30 day lines
    # ts.get_hist_data(str(code), ktype='M').to_csv(path+"stock_data/"+str(code)+'_month.csv')