Python+Tkinter實(shí)現(xiàn)股票K線圖的繪制
在前面的文章中,我們一起學(xué)習(xí)了如何通過 Python 抓取東方財(cái)富網(wǎng)的實(shí)時(shí)股票數(shù)據(jù)以及如何制作成 Tkinter GUI 程序,鏈接如下
用 Python 爬取股票實(shí)時(shí)數(shù)據(jù)
Tkinter制作股票數(shù)據(jù)抓取小程序,有點(diǎn)秀!
今天我們就在這個(gè)基礎(chǔ)上,在 Tkinter 程序中繪制 K 線圖,一起來(lái)看看吧

子窗口
我們今天的整體代碼還是基于上次的 Tkinter 股票程序,在主類 MainCreator 下面創(chuàng)建一個(gè)函數(shù) create_subwindow
????def?create_subwindow(self):
????????t?=?ttk.Toplevel()
????????t.wm_title("K線圖")
????????sub_window?=?SubWindiw()
????????sub_window.subWindow(t)
我們這里使用 Toplevel 來(lái)創(chuàng)建子窗口,再來(lái)看看子窗口相關(guān)的代碼
class?SubWindiw: ????def?__init__(self): ????????pass ????def?all_files(self): ??????... ????def?getDate(self): ??????... ????def?subWindow(self,?root_frame): ??????... ????def?go(self): ??????...
對(duì)于子窗口代碼,subWindow 函數(shù)是子窗口的主要函數(shù)
子窗口框架
子窗口主要代碼如下
????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)建標(biāo)簽‘開始日期' ????????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)建標(biāo)簽‘結(jié)束日期' ????????date_end_label?=?ttk.Label(date_frame,?text='結(jié)束日期') ????????date_end_label.pack(side=LEFT) ????????#?創(chuàng)建結(jié)束日期代碼輸入框 ????????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)建查詢按鈕并設(shè)置功能 ????????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)
主要還是分三個(gè)區(qū)域,日期選擇區(qū)域,股票文件選擇區(qū)域以及K線圖展示區(qū)域
還有兩個(gè)功能性函數(shù) 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("股票文件為空",?"請(qǐng)選擇股票數(shù)據(jù)文件!")
????????????return
????????stock_data?=?pd.read_csv(gupiao_file)
????????data?=?stock_data.loc[:,?['時(shí)間',?'開盤價(jià)',
??????????????????????????????????'收盤價(jià)',?'最高價(jià)',?'最低價(jià)',?'成交量(手)']]??#?:取所有行數(shù)據(jù),后面取date列,open列等數(shù)據(jù)
????????data?=?data.rename(columns={'時(shí)間':?'Date',?'開盤價(jià)':?'Open',
????????????????????????????????????'收盤價(jià)':?'Close',?'最高價(jià)':?'High',?'最低價(jià)':?'Low',
????????????????????????????????????'成交量(手)':?'Volume'})??#?更換列名,為后面函數(shù)變量做準(zhǔn)備
????????data.set_index('Date',?inplace=True)??#?設(shè)置date列為索引,覆蓋原來(lái)索引,這個(gè)時(shí)候索引還是 object 類型,就是字符串類型。
????????#?將object類型轉(zhuǎn)化成 DateIndex 類型,pd.DatetimeIndex 是把某一列進(jìn)行轉(zhuǎn)換,同時(shí)把該列的數(shù)據(jù)設(shè)置為索引 index。
????????data.index?=?pd.DatetimeIndex(data.index)
????????data?=?data.sort_index(ascending=True)??#?將時(shí)間順序升序,符合時(shí)間序列
????????data?=?data[data.index?<?end_date][data.index?>?start_date]
????????if?not?data.values.any():
????????????print("股票數(shù)據(jù)為空")
????????????messbox.showerror("股票數(shù)據(jù)為空",?"請(qǐng)選擇合理的時(shí)間!")
????????????return
????????my_color?=?mpf.make_marketcolors(up='r',
?????????????????????????????????????????down='g',
?????????????????????????????????????????edge='inherit',
?????????????????????????????????????????wick='inherit',
?????????????????????????????????????????volume='inherit')
????????#?設(shè)置圖表的背景色
????????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)??#?設(shè)置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 來(lái)繪制K線圖
首先是通過 Pandas 來(lái)處理數(shù)據(jù),把我們爬取到的數(shù)據(jù)處理成 mplfinance 需要的格式,如下

接下來(lái)只需要調(diào)用plot函數(shù)即可
self.fig,?self.axlist?=?mpf.plot(data,?style=my_style,?type='candle', ?????????????????????????????????????????mav=(5,?10,?20),?volume=True,?show_nontrading=False,?returnfig=True)
到此這篇關(guān)于Python+Tkinter實(shí)現(xiàn)股票K線圖的繪制的文章就介紹到這了,更多相關(guān)Python Tkinter股票K線圖內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python機(jī)器學(xué)習(xí)樸素貝葉斯算法及模型的選擇和調(diào)優(yōu)詳解
這篇文章主要為大家介紹了python機(jī)器學(xué)習(xí)樸素貝葉斯及模型的選擇和調(diào)優(yōu)示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步2021-11-11
2024最新Pytorch安裝圖文教程(Anaconda+GPU)
本文詳細(xì)介紹了如何使用Anaconda創(chuàng)建虛擬環(huán)境,并在該環(huán)境中安裝CUDA和cuDNN,最后安裝支持GPU的PyTorch,本文通過圖文并茂的形式給大家介紹的非常詳細(xì),感興趣的朋友一起看看吧2024-11-11
python 利用matplotlib在3D空間繪制二次拋物面的案例
這篇文章主要介紹了python 利用matplotlib在3D空間繪制二次拋物面的案例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧2021-02-02
Python實(shí)現(xiàn)PS圖像調(diào)整之對(duì)比度調(diào)整功能示例
這篇文章主要介紹了Python實(shí)現(xiàn)PS圖像調(diào)整之對(duì)比度調(diào)整功能,結(jié)合實(shí)例形式分析了Python實(shí)現(xiàn)PS圖像對(duì)比度調(diào)整的原理、實(shí)現(xiàn)方法及相關(guān)操作技巧,需要的朋友可以參考下2018-01-01
新手該如何學(xué)python怎么學(xué)好python?
怎么學(xué)好python?怎么靈活應(yīng)用python?2008-10-10
python動(dòng)態(tài)進(jìn)度條的實(shí)現(xiàn)代碼
有時(shí)候我們需要使用print打印工作進(jìn)度,正常使用print函數(shù)會(huì)導(dǎo)致刷屏的現(xiàn)象,本文通過實(shí)例代碼給大家介紹python動(dòng)態(tài)進(jìn)度條的實(shí)現(xiàn)方法,感興趣的朋友跟隨小編一起看看吧2019-07-07

