|
|
|
@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
# 提取综合排名前20的股票代码列表
|
|
|
|
|
|
|
|
import fun
|
|
|
|
|
|
|
|
r = fun.Fr()
|
|
|
|
|
|
|
|
c = r[0]
|
|
|
|
|
|
|
|
codelist = list(c.index[0:20])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 构建排名前20个股票代码的查询字符（连接）
|
|
|
|
|
|
|
|
codelist_str = ','.join(codelist)
|
|
|
|
|
|
|
|
print(codelist_str)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 批量获取20个股票代码交易数据stkdata，并导出到Excel
|
|
|
|
|
|
|
|
import tushare as ts
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 设置 Tushare API token
|
|
|
|
|
|
|
|
ts.set_token('64532b1c03637bc0c3ac92931a5d1b53cfaf75de87c22dfdc70ca6a0')
|
|
|
|
|
|
|
|
pro = ts.pro_api()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 获取股票交易数据
|
|
|
|
|
|
|
|
stkdata = pro.daily(ts_code=codelist_str, start_date='20170101', end_date='20171231')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 按股票代码和交易日期排序
|
|
|
|
|
|
|
|
stkdata = stkdata.sort_values(['ts_code', 'trade_date'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 重新设置索引
|
|
|
|
|
|
|
|
stkdata.index = range(len(stkdata))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# 导出到Excel
|
|
|
|
|
|
|
|
stkdata.to_excel('stkdata.xlsx')
|
