python tushare for echarts 生成符合要求的 json

pip install tushare ;

stock2json.py

# coding: utf-8
import os, sys
import pandas as pd
import tushare as ts


if len(sys.argv) ==2:
    code = sys.argv[1]
else:
    print('usage: python stock2json.py stockcode ')
    sys.exit(1)

if len(code) !=6:
    print('stock code length: 6')
    sys.exit(2)

# help(ts.get_k_data) 了解参数
k_data = ts.get_k_data(code)
# 取样 2018年以后的收盘价
df = k_data[ k_data['date'] >'2018-01-01']

f2 = code +'.json'
fp2 = open(f2,'w')
fp2.write('[')
i = 0
n = df.shape[0] # 行数
for idx,row in df.iterrows():
    i +=1
    date,open,close,high,low,volume = row[0:6]
    if i < n:
        ss = "['%s',%s,%s,%s,%s],\n" % (date,open,close,low,high)
    else:
        ss = "['%s',%s,%s,%s,%s]]" % (date,open,close,low,high)
    fp2.write(ss)
#
fp2.close()
print(" {0} rows".format(i))

运行 python stock2json.py 600030