当前位置 博文首页 > LY的博客:量化策略多因子选股之抓取数据并保存到本机MYSQL数据

    LY的博客:量化策略多因子选股之抓取数据并保存到本机MYSQL数据

    作者:[db:作者] 时间:2021-08-10 09:54

    数据来自于tushare

    使用了sqlalchemy 包的引擎来保存数据与SQL

    
    
    # coding=utf-8
    import pandas as pd
    import tushare as ts
    import sys
    from sqlalchemy import create_engine
    
    reload(sys)
    sys.setdefaultencoding("utf-8")
    
    
    zz500=ts.get_zz500s()
    hs300=ts.get_hs300s()
    
    zz500_code= zz500['code']
    hs300_code= hs300['code']
    
    get_industry= ts.get_industry_classified()
    
    print get_industry
    # 得到sina行业分类
    get_industry_hs300= get_industry[get_industry['code'].isin(hs300_code) ]
    get_industry_zz500= get_industry[get_industry['code'].isin(zz500_code) ]
    
    
    #得到2017年第四季度的主报数据
    df1=get_report=ts.get_report_data(2017,4)
    
    
    #得到2017年第四季度的盈利能力数据
    df2=ts.get_profit_data(2017,4)
    
    
    #得到2017年第四季度的成长能力数据
    df3=ts.get_growth_data(2017,4)
    
    
    engine = create_engine('mysql://root:********@127.0.0.1/stock_report?charset=utf8')
    
    df1.to_sql('get_report',engine)
    df2.to_sql('get_report2',engine)
    df3.to_sql('get_report3',engine)
    
    frames = [get_industry_hs300, get_industry_zz500]
    result = pd.concat(frames)
    result.to_sql('hangye',engine)
    
    

    数据预览:

    cs