将matplotlib绘制的股票的图表镶嵌入tkinter做的界面中.zip

      # -*- coding: utf-8 -*-
import tkinter as Tk
import pandas as pd
import tushare as ts


import matplotlib.pyplot as plt
from mpl_finance import candlestick2_ochl
import matplotlib.ticker as ticker
#import numpy as np
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk
from matplotlib.backend_bases import key_press_handler
#from matplotlib.figure import Figure
import tkinter 
begin_time = '2017-07-01'
end_time = '2019-09-01'
code = "000001"
df = ts.get_hist_data(code, start=begin_time)#, end=end_time
df.to_csv('000001.csv')
df = df.sort_index(0)
df_idx = df.index.values
root = Tk.Tk()
root.minsize(900,700)

#    df.plot()

fig1, ax = plt.subplots(figsize=(15, 6)) # 设置图片大小。

# https://matplotlib.org/api/finance_api.html#module-matplotlib.finance
# matplotlib.finance.candlestick2_ochl(ax, opens, closes, highs, lows, width=4, colorup='r', colordown='g', alpha=0.75)
candlestick2_ochl(ax = ax, 
                 opens=df["open"].values, closes=df["close"].values,
                 highs=df["high"].values, lows=df["low"].values, 
                 width=0.75, colorup='r', colordown='g', alpha=0.75)
df["5d"] = df["close"].rolling(window = 5).mean() #周线
df["10d"] =df["close"].rolling(window = 10).mean() #半月线
df["20d"] = df["close"].rolling(window = 20).mean() #月线
df["60d"] = df["close"].rolling(window = 60).mean() #季度线
df[["close","5d","10d","20d","60d",]].plot(ax=ax,figsize=(20,10), grid=True)
ax.xaxis.set_major_locator(ticker.MaxNLocator(20))
# 设置自动格式化时间。
def mydate_formatter(x,pos):
    try:
        return df_idx[int(x)]
    except IndexError:
        return ''
ax.xaxis.set_major_formatter(ticker.FuncFormatter(mydate_formatter))

plt.setp(plt.gca().get_xticklabels(), rotation=45, horizontalalignment='right')
ax.grid(True)
plt.title(code)
fig2, ax2 = plt.subplots(figsize=(15, 6)) # 设置图片大小。
df["volume5"]=df["volume"].rolling(window = 5).mean() #周线
df["volume10"]=df["volume"].rolling(window = 10).mean() #周线
df["volume30"]=df["volume"].rolling(window = 30).mean() #周线
df[["volume","volume5","volume10","volume30"]].plot(ax=ax2,figsize=(20,10), grid=True)

#plt.show()


canvas = FigureCanvasTkAgg(fig1, master=root)
canvas.draw()  # 注意show方法已经过时了,这里改用draw
canvas.get_tk_widget().pack(side=tkinter.TOP,  # 上对齐
                            fill=tkinter.BOTH,  # 填充方式
                            expand=tkinter.YES)  # 随窗口大小调整而调整
# matplotlib的导航工具栏显示上来(默认是不会显示它的)
toolbar = NavigationToolbar2Tk(canvas, root)
toolbar.update()
canvas._tkcanvas.pack(side=tkinter.TOP,  # get_tk_widget()得到的就是_tkcanvas
                      fill=tkinter.BOTH,
                      expand=tkinter.YES)

canvas = FigureCanvasTkAgg(fig2, master=root)
canvas.draw()  # 注意show方法已经过时了,这里改用draw
canvas.get_tk_widget().pack(side=tkinter.TOP,  # 上对齐
                            fill=tkinter.BOTH,  # 填充方式
                            expand=tkinter.YES)  # 随窗口大小调整而调整
toolbar = NavigationToolbar2Tk(canvas, root)
toolbar.update()
canvas._tkcanvas.pack(side=tkinter.TOP,  # get_tk_widget()得到的就是_tkcanvas
                      fill=tkinter.BOTH,
                      expand=tkinter.YES)
main_frame = Tk.Frame(root)
main_frame.pack()

menu_frame = Tk.Frame(root)
menu_frame.pack()

tab_frame = Tk.Frame(root)
tab_frame.pack()

chart_frame = Tk.Frame(root)
chart_frame.pack()

statusbar_frame = Tk.Frame(root)
statusbar_frame.pack(side=Tk.BOTTOM, fill=Tk.BOTH, expand=True)

# functions ---------------------------------------------
def on_key_event(event):
    """键盘事件处理"""
    print("你按了%s" % event.key)
    key_press_handler(event, canvas, toolbar)


# 绑定上面定义的键盘事件处理函数
canvas.mpl_connect('key_press_event', on_key_event)

def NewFile():
    print ("New File!")
def OpenFile():
    name = askopenfilename()
    print (name)
def About():
    print ("This is a simple example of a menu")
# ----------------------------------------------------------

# Menu --------------------------------------------------------
menu = Tk.Menu(menu_frame)
root.config(menu=menu)
filemenu = Tk.Menu(menu)
menu.add_cascade(label="File", menu=filemenu)
filemenu.add_command(label="New", command=NewFile)
filemenu.add_command(label="Open...", command=OpenFile)
filemenu.add_separator()
filemenu.add_command(label="Exit", command=root.destroy)

helpmenu = Tk.Menu(menu)
menu.add_cascade(label="Help", menu=helpmenu)
helpmenu.add_command(label="About...", command=About)
root.mainloop()