python numpy+matplotlib 画股票k线图

pylist 1191

python numpy+matplotlib 画股票k线图

# -- coding: utf-8 --
import requests
import numpy as np   
from matplotlib import pyplot as plt   
from matplotlib import animation

fig = plt.figure(figsize=(8,6), dpi=72,facecolor="white")
axes = plt.subplot(111)
axes.set_title('Shangzheng')
axes.set_xlabel('time')
line, = axes.plot([], [], linewidth=1.5, linestyle='-')
alldata = []

def dapan(code):
    url = 'http://hq.sinajs.cn/?list='+code
    r = requests.get(url)
    data = r.content[21:-3].decode('gbk').encode('utf8').split(',')
    alldata.append(data[3])
    axes.set_ylim(float(data[5]), float(data[4]))
    return alldata

def init():
    line.set_data([], [])
    return line

def animate(i): 
    axes.set_xlim(0, i+10)
    x = range(i+1)
    y = dapan('sh000001')
    line.set_data(x, y)
    return line

anim=animation.FuncAnimation(fig, animate, init_func=init,  frames=10000, interval=5000)

plt.show()

https://raw.githubusercontent.com/lugeek/matplotlib-stock/master/dapan.py 22

登录发表评论