利用Tushare进行股票因子计算

开发背景

笔者tushare ID：414988

一、获取沪深300成分股

基本思路：

1. 先获取全部股票
2. 获取沪深300股票代码
3. 根据沪深300股票代码，提取对应的ts代码
4. 根据ts代码获取沪深300股票列表

1. 获取全部股票代码

stock_list = pro.stock_basic(exchange='',list_status='L', fields='ts_code,symbol,name,area,industry,market,exchange')
stock_list

ts_code	symbol	name	area	industry	market	exchange
0	000001.SZ	000001	平安银行	深圳	银行	主板	SZSE
1	000002.SZ	000002	万科A	深圳	全国地产	主板	SZSE
2	000004.SZ	000004	国华网安	深圳	软件服务	主板	SZSE
3	000005.SZ	000005	世纪星源	深圳	环境保护	主板	SZSE
4	000006.SZ	000006	深振业A	深圳	区域地产	主板	SZSE
...	...	...	...	...	...	...	...
4231	688777.SH	688777	中控技术	浙江	软件服务	科创板	SSE
4232	688788.SH	688788	科思科技	深圳	通信设备	科创板	SSE
4233	688819.SH	688819	天能股份	浙江	电气设备	科创板	SSE
4234	688981.SH	688981	中芯国际	上海	半导体	科创板	SSE
4235	689009.SH	689009	九号公司-UWD	北京	专用机械	CDR	SSE
4236 rows × 7 columns

2. 获取沪深300成分股代码

hs300s = ts.get_hs300s() 
hs300s

	date	code	name	weight
0	2021-03-31	600000	浦发银行	0.68
1	2021-03-31	600004	白云机场	0.08
2	2021-03-31	600009	上海机场	0.29
3	2021-03-31	600010	包钢股份	0.19
4	2021-03-31	600011	华能国际	0.10
...	...	...	...	...
295	2021-03-31	300498	温氏股份	0.40
296	2021-03-31	300529	健帆生物	0.16
297	2021-03-31	300601	康泰生物	0.25
298	2021-03-31	300628	亿联网络	0.10
299	2021-03-31	300676	华大基因	0.13
300 rows × 4 columns

hs300_symbols = hs300s.loc[:]['code']
hs300_symbols

0      600000
1      600004
2      600009
3      600010
4      600011
        ...  
295    300498
296    300529
297    300601
298    300628
299    300676
Name: code, Length: 300, dtype: object

3. 根据沪深300股票代码，提取对应的ts代码

# 获取沪深300的所有股票的ts代码
hs300_code_list