[015量化交易] python 通过tushare将股票代码和股票名称存入json文件中

[015量化交易] python 通过tushare将股票代码和股票名称存入json文件中

#! /usr/bin/env python
# -*- encoding: utf-8 -*-
import jsonimport tushare as tsdef tushare_to_json():# 设置tokentoken = ' '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_date0  000001.SZ  000001  平安银行   深圳       银行     主板  199104031  000002.SZ  000002   万科A   深圳     全国地产     主板  199101292  000004.SZ  000004  国农科技   深圳     生物制药     主板  199101143  000005.SZ  000005  世纪星源   深圳     环境保护     主板  199012104  000006.SZ  000006  深振业A   深圳     区域地产     主板  19920427"""codes = df.ts_code.valuesnames = df.name.valuesstock = 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)