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

#! /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)