tushare-学习笔记002-获取基本数据
github 链接
先获取 “股票代码表”
我这里是使用 mongodb 存储的。
项目有结构不方便放上来，这里是代码片段。具体请参见 github 仓库
def fetch_code():
"""
从接口获取代码，存入数据库，返回结构样例
ts_code symbol name area industry list_date
0 000001.SZ 000001 平安银行 深圳 银行 19910403
1 000002.SZ 000002 万科A 深圳 全国地产 19910129
3541 603936.SH 603936 博敏电子 广东 元器件 20151209
3542 603937.SH 603937 丽岛新材 江苏 铝 20171102
:return:
"""
# 查询当前所有正常上市交易的股票列表
datas = ts_pro.query('stock_basic', exchange='', list_status='L', fields='ts_code,symbol,name,area,industry,list_date')
dataArr = datas.to_dict('records')
# 创建集合，相当于表
t_codes = db['codes']
t_codes.drop()
# 插入获取的代码
result = t_codes.insert_many(dataArr)
if len(result.inserted_ids) != len(dataArr):
assert False
print('插入代码表数据失败.请检查。')
return
else:
print("插入代码表数据成功")
获取交易日历
def fetch_calender():
"""
从接口获取有效交易日信息，存入数据库，返回结构样例
exchange cal_date is_open
0 SSE 20180101 0
1 SSE 20180102 1
2 SSE 20180103 1
"""
calender = ts_pro.trade_cal(start_date='20190101', end_date='20191231')
# 创建集合，相当于表
t_calender = db['calender']
dataArr = calender.to_dict('records')
result = t_calender.insert_many(dataArr)
if len(result.inserted_ids) != len(dataArr):
assert False
print('插入代码表数据失败.请检查。')
return
else:
print("插入代码表数据成功")
获取历史数据
代码有些多，看 github 吧
每个股票代码作为一个集合保存数据。从数据库中取出代码，然后遍历获取数据
def get_code(market):
"""
获取股票代码表
market [in] string
SH 上海,SZ 深圳,ALL 所有市场
output:
{'ts_code': '600000.SH', 'symbol': '600000', 'name': '浦发银行', 'list_date': '19991110'}
:return:
"""
# 打开代码表集合
t_codes = db['codes']
if market == 'ALL':
codes = list(t_codes.find({}, {"_id": 0, "ts_code": 1, "symbol": 1, "name": 1, "industry": 1, "list_date": 1}))
else:
codes = list(t_codes.find({"ts_code": {"$regex": ".{}".format(market)}},
{"_id": 0, "ts_code": 1, "symbol": 1, "name": 1, "industry": 1, "list_date": 1}))
return codes
到这里基本数据就准备完毕了，在下一个笔记就可以选股操作了。