#! /usr/bin/env python
# -*- encoding: utf-8 -*-
import json
import tushare as ts
def tushare_to_json():
# 设置token
token = ' '
pro = ts.pro_api(token) # 初始化pro接口
index = {'上证综指': 'sh.000001',
'深证成指': 'sz.399001',
'沪深300': 'sz.000300',
'创业板指': 'sz.399006',
'上证50': 'sh.000016',
'中证500': 'sh.000905',
'中小板指': 'sz.399005',
'上证180': 'sh.000010'}
df = pro.stock_basic(exchange='', list_status='L')
print(df.head())
"""
ts_code symbol name area industry market list_date
0 000001.SZ 000001 平安银行 深圳 银行 主板 19910403
1 000002.SZ 000002 万科A 深圳 全国地产 主板 19910129
2 000004.SZ 000004 国农科技 深圳 生物制药 主板 19910114
3 000005.SZ 000005 世纪星源 深圳 环境保护 主板 19901210
4 000006.SZ 000006 深振业A 深圳 区域地产 主板 19920427
"""
codes = df.ts_code.values
names = df.name.values
stock = dict(zip(names, codes))
# print(stock)
# {'平安银行': '000001.SZ', '万科A': '000002.SZ', ......}
stock_index = dict([('指数', index), ('股票', stock)])
# print(stock_index)
# stock_index = dict(stock, **index) # 合并指数和个股成一个字典
# dump: 将数据写入json文件中
with open("stock_pool.json", "w", encoding='utf-8') as f:
json.dump(stock_index, f, ensure_ascii=False, indent=4)
[015量化交易] python 通过tushare将股票代码和股票名称存入json文件中
最新推荐文章于 2024-04-24 09:07:52 发布