Python+Tkinter實現股票K線圖的繪制

更新時間：2022年08月25日 10:37:16 作者：周蘿卜

K線圖又稱蠟燭圖，常用說法是“K線”。K線是以每個分析周期的開盤價、最高價、最低價和收盤價繪制而成。本文將利用Python+Tkinter實現股票K線圖的繪制，需要的可以參考一下

子窗口

我們今天的整體代碼還是基于上次的 Tkinter 股票程序，在主類 MainCreator 下面創(chuàng)建一個函數 create_subwindow

????def?create_subwindow(self):
????????t?=?ttk.Toplevel()
????????t.wm_title("K線圖")
????????sub_window?=?SubWindiw()
????????sub_window.subWindow(t)

我們這里使用 Toplevel 來創(chuàng)建子窗口，再來看看子窗口相關的代碼

class?SubWindiw:
????def?__init__(self):
????????pass
????def?all_files(self):
??????...
????def?getDate(self):
??????...
????def?subWindow(self,?root_frame):
??????...
????def?go(self):
??????...

對于子窗口代碼，subWindow 函數是子窗口的主要函數

子窗口框架

子窗口主要代碼如下

????def?subWindow(self,?root_frame):
????????file_list?=?self.all_files()

????????#?創(chuàng)建主框架
????????main_frame?=?ttk.Frame(root_frame)
????????main_frame.pack()

????????#?在主框架下創(chuàng)建股票日期輸入框子框架
????????date_frame?=?ttk.Frame(main_frame,?relief=tix.SUNKEN)
????????date_frame.pack(fill=X,?side=TOP)
????????#?創(chuàng)建標簽‘開始日期'
????????date_start_label?=?ttk.Label(date_frame,?text='開始日期')
????????date_start_label.pack(side=LEFT)
????????#?創(chuàng)建開始日期代碼輸入框
????????re_date?=?self.getDate()
????????self.de_start?=?ttk.DateEntry(date_frame,?startdate=re_date)
????????self.de_start.pack(side=LEFT,?padx=4,?pady=4)
????????#?創(chuàng)建標簽‘結束日期'
????????date_end_label?=?ttk.Label(date_frame,?text='結束日期')
????????date_end_label.pack(side=LEFT)
????????#?創(chuàng)建結束日期代碼輸入框
????????self.de_end?=?ttk.DateEntry(date_frame)
????????self.de_end.pack(side=LEFT,?padx=4,?pady=4)

????????#?在主框架下創(chuàng)建查詢按鈕子框架
????????search_frame?=?ttk.Frame(main_frame,?relief=tix.SUNKEN)
????????search_frame.pack(fill=X,?side=TOP)
????????#?文件選擇框
????????search_label?=?ttk.Label(search_frame,?text='選擇股票文件')
????????search_label.pack(side=LEFT)
????????values?=?file_list
????????self.cbo?=?ttk.Combobox(
????????????master=search_frame,
????????????values=values,
????????????width=45,
????????)
????????self.cbo.pack(side=LEFT,?expand=YES,?pady=5,?fill=X)

????????#?占位?check?button
????????cb?=?ttk.Checkbutton(search_frame,?text="占位")
????????#?cb.pack(side=LEFT,?padx=5,?fill=X)

????????#?創(chuàng)建查詢按鈕并設置功能
????????stock_find?=?ttk.Button(search_frame,?text='查詢',?command=self.go)
????????stock_find.pack(pady=4,?expand=YES,?side=LEFT)

????????#?創(chuàng)建股票圖形輸出框架
????????self.stock_graphics?=?tk.Frame(root_frame,?relief=tix.RAISED)
????????self.stock_graphics.pack(expand=1,?fill=tix.BOTH,?anchor=tix.CENTER)

主要還是分三個區(qū)域，日期選擇區(qū)域，股票文件選擇區(qū)域以及K線圖展示區(qū)域

還有兩個功能性函數 all_files 和 getDate

all_files

????def?all_files(self):
????????file_list?=?[]
????????for?root,?dirs,?files?in?os.walk("./"):
????????????for?f?in?files:
????????????????if?'csv'?in?f?and?'days'?in?f:
????????????????????file_list.append(f)
????????return?file_list

getDate

????def?getDate(self):
????????today?=?datetime.datetime.now()
????????offset?=?datetime.timedelta(days=-1)
????????re_date?=?(today?+?offset).strftime('%Y/%m/%d')
????????return?today?+?offset

繪制K線圖

最后我們看看繪制K線圖的代碼

????def?go(self):
????????code_name?=?self.cbo.get()
????????start_date?=?self.de_start.entry.get()
????????end_date?=?self.de_end.entry.get()
????????gupiao_file?=?self.cbo.get()
????????if?not?gupiao_file:
????????????messbox.showerror("股票文件為空",?"請選擇股票數據文件！")
????????????return
????????stock_data?=?pd.read_csv(gupiao_file)
????????data?=?stock_data.loc[:,?['時間',?'開盤價',
??????????????????????????????????'收盤價',?'最高價',?'最低價',?'成交量（手）']]??#?：取所有行數據，后面取date列，open列等數據
????????data?=?data.rename(columns={'時間':?'Date',?'開盤價':?'Open',
????????????????????????????????????'收盤價':?'Close',?'最高價':?'High',?'最低價':?'Low',
????????????????????????????????????'成交量（手）':?'Volume'})??#?更換列名，為后面函數變量做準備
????????data.set_index('Date',?inplace=True)??#?設置date列為索引，覆蓋原來索引,這個時候索引還是 object 類型，就是字符串類型。
????????#?將object類型轉化成 DateIndex 類型，pd.DatetimeIndex 是把某一列進行轉換，同時把該列的數據設置為索引 index。
????????data.index?=?pd.DatetimeIndex(data.index)
????????data?=?data.sort_index(ascending=True)??#?將時間順序升序，符合時間序列
????????data?=?data[data.index?<?end_date][data.index?>?start_date]

????????if?not?data.values.any():
????????????print("股票數據為空")
????????????messbox.showerror("股票數據為空",?"請選擇合理的時間！")
????????????return
????????my_color?=?mpf.make_marketcolors(up='r',
?????????????????????????????????????????down='g',
?????????????????????????????????????????edge='inherit',
?????????????????????????????????????????wick='inherit',
?????????????????????????????????????????volume='inherit')
????????#?設置圖表的背景色
????????my_style?=?mpf.make_mpf_style(marketcolors=my_color,
??????????????????????????????????????figcolor='#002B36',
??????????????????????????????????????facecolor='#002B36',
??????????????????????????????????????edgecolor='w',
??????????????????????????????????????#?gridcolor='(0.82,?0.83,?0.85)',
??????????????????????????????????????rc={'font.family':?'SimHei',
??????????????????????????????????????????'xtick.labelcolor':?'white',
??????????????????????????????????????????'ytick.labelcolor':?'white',
??????????????????????????????????????????'axes.labelcolor':?'white',
??????????????????????????????????????????})
????????self.fig,?self.axlist?=?mpf.plot(data,?style=my_style,?type='candle',
?????????????????????????????????????????mav=(5,?10,?20),?volume=True,?show_nontrading=False,?returnfig=True)
????????canvas?=?FigureCanvasTkAgg(self.fig,?master=self.stock_graphics)??#?設置tkinter繪制區(qū)
????????if?len(self.stock_graphics.winfo_children())?==?2:
????????????self.stock_graphics.winfo_children()[0].destroy()
????????canvas.draw()

????????canvas._tkcanvas.pack(side=BOTTOM,?fill=BOTH,?expand=1)

我們通過 Matplotlib + mplfinance 來繪制K線圖

首先是通過 Pandas 來處理數據，把我們爬取到的數據處理成 mplfinance 需要的格式，如下

接下來只需要調用plot函數即可

self.fig,?self.axlist?=?mpf.plot(data,?style=my_style,?type='candle',
?????????????????????????????????????????mav=(5,?10,?20),?volume=True,?show_nontrading=False,?returnfig=True)

到此這篇關于Python+Tkinter實現股票K線圖的繪制的文章就介紹到這了,更多相關Python Tkinter股票K線圖內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: