选择股票池

获取沪深300股票池

import tushare as ts
name = ts.get_hs300s()
name_list = list(name.code)
print(name_list)
right_names = map(lambda code: code+'.XSHG' if code.startswith('6') else code+ '.XSHE', name_list)
print right_names

# 获取创业板股票池
ts.get_gem_classified()
# 获取中小板股票池
ts.get_sme_classified()
# 获取中证500股票池
ts.get_zz500s()

获取不同分类的股票池

#coding:utf-8
from fxdayu_data import DataAPI
import tushare as ts
import pandas as pd
classified = ts.get_industry_classified()
classified.to_excel('classified.xlsx')
data = pd.read_excel('classified.xlsx')
cond = data.c_name==u'家电行业'
codes = data[cond].code.values
name_list = []
for c in codes:
    right_code = '%06.f'%c
    name_list.append(right_code)
print name_list
right_names = map(lambda code: code+'.XSHG' if code.startswith('6') else code+ '.XSHE', name_list)