使用Python實現(xiàn)從零開始打造一個三維繪圖系統(tǒng)
框架
本文的目標是實現(xiàn)一個下圖所示的系統(tǒng)，通過指定x,y,z的表達式，以實現(xiàn)三維繪圖的目的。這個需求其實此前也實現(xiàn)過，見此文，但其內(nèi)容比較駁雜，并不利于快速實現(xiàn)，相比之下，本文直指核心問題，代碼也進一步精簡，更加適合學習。
為此，需要經(jīng)歷三個過程
實現(xiàn)將matplotlib的繪圖窗口嵌入到tkinter的窗口中
實現(xiàn)對x,y,z數(shù)據(jù)的讀取
實現(xiàn)三維繪圖
關(guān)于第一個問題，matplotlib中的圖窗和工具欄分別是FigureCanvasTkAgg和NavigationToolbar2Tk對象，只需將二者押入到tkinter的布局窗口，即可實現(xiàn)二者的融合。
對于第二個問題，只需使用eval函數(shù)便可輕松實現(xiàn)，唯一的難點，則是x,y,z三者之間的聯(lián)系。
而只要第二個問題解決，那么第三個問題也就不是問題了，只需通過一個繪圖函數(shù)，便可輕松解決。
下面代碼給出了實現(xiàn)這個需求需要導入的模塊，以及整個系統(tǒng)的基本框架
import tkinter as tk
import tkinter.ttk as ttk
import matplotlib as mpl
mpl.use('TkAgg')
from matplotlib.backends.backend_tkagg import (
FigureCanvasTkAgg, NavigationToolbar2Tk)
from matplotlib.figure import Figure
import numpy as np
class DarwSystem():
def __init__(self):
self.root = tk.Tk()
self.root.title("數(shù)據(jù)展示工具")
self.data = {}
self.setFrmCtrl()
self.setFrmFig()
self.root.mainloop()
def setFrmCtrl(self): pass
def setFrmFig(self): pass
def setFrmAxis(self, frm, flag): pass
def setCtrlButtons(self, frm): pass
def readEntrys(self): pass
def btnDrawImg(self): pass其中，各函數(shù)含義如下
- setFrmCtrl 用于右側(cè)工具欄的組件布局
- setFrmFig 用于左側(cè)繪圖窗口的嵌入，解決第一個問題
- setFrmAxis 工具欄中的坐標輸入框相似度很高，可以封裝成專門的函數(shù)
- readEntrys 讀取輸入窗口的數(shù)據(jù)，解決第二個問題
- btnDrawImg 繪圖函數(shù)，解決第三個問題
下面逐個講解每個函數(shù)的實現(xiàn)過程，這些函數(shù)都是DataSystem類的成員，所以在書寫時注意縮進。
布局
考慮matplotlib繪圖窗口的嵌入是一個相對較難的問題，故而先行實現(xiàn)setFrmFig函數(shù)
def setFrmFig(self):
frmFig = ttk.Frame(self.root)
frmFig.pack(side=tk.LEFT,fill=tk.BOTH,expand=tk.YES)
self.fig = Figure()
self.canvas = FigureCanvasTkAgg(self.fig,frmFig)
self.canvas.get_tk_widget().pack(
side=tk.TOP,fill=tk.BOTH,expand=tk.YES)
self.toolbar = NavigationToolbar2Tk(self.canvas,frmFig,
pack_toolbar=False)
self.toolbar.update()
self.toolbar.pack(side=tk.RIGHT)
其中self.fig就是繪圖窗口，后續(xù)若要畫圖，都要在這里設置坐標軸。canvas即繪圖畫布，toolbar則是畫布下方的工具欄。在這里，可以將二者直接視為平平無奇的痛苦inter組件。
另一個和布局相關(guān)的函數(shù)是setFrmCtrl，其中又將x,y,z三個坐標軸的設置封裝成了一個函數(shù)，以簡化代碼。唯一的繪圖按鈕，綁定了btnDrawImg函數(shù)，將在后續(xù)實現(xiàn)。
def setFrmCtrl(self):
frmCtrl = ttk.Frame(self.root,width=320)
frmCtrl.pack(side=tk.RIGHT, fill=tk.Y)
frm = ttk.Frame(frmCtrl, width=320)
frm.pack(side=tk.TOP, fill=tk.X)
ttk.Button(frm, text="繪圖",width=5,
command=self.btnDrawImg).pack(side=tk.LEFT)
self.entrys = {}
for flag in 'xyz':
frm = ttk.Frame(frmCtrl)
frm.pack(side=tk.TOP, fill=tk.X)
self.setFrmAxis(frm, flag)
def setFrmAxis(self, frm, flag):
tk.Label(frm, text=flag).pack(side=tk.LEFT)
self.entrys[flag] = tk.Entry(frm)
self.entrys[flag].pack(side=tk.LEFT, fill=tk.X)
繪圖函數(shù)
繪圖包括兩個過程，一是讀取輸入內(nèi)容，將其轉(zhuǎn)換可供繪圖的數(shù)據(jù)；二則是將這些數(shù)據(jù)繪制成圖。
首先，第一步實現(xiàn)如下
def readEntrys(self):
self.data = {}
for flag in 'xyz':
label = self.entrys[flag].get()
if label=="": continue
self.data[flag] = eval(self.entrys[flag].get())
if flag =='x' : x = self.data['x']
elif flag =='y' : y = self.data['y']
首先，定義一個全局字典self.data用于繪圖。
然后簡單起見，用eval函數(shù)直接讀取python表達式。但需要注意的是，由于循環(huán)順序是x→y→z，所以x的表達式中不能出現(xiàn)y,z；y的表達式中只能出現(xiàn)x；z的表達式中x,y都可以出現(xiàn)。
最后是繪圖，這里需要實現(xiàn)給定x,y和x,y,z這兩種情況的繪圖代碼，前者是二維坐標系，后者是三維的。而canvas的繪圖邏輯與plt如出一轍，都是先創(chuàng)建一個坐標軸，然后在坐標軸上繪圖，最后也只需把plt.draw改為canvas.draw，代碼如下
def btnDrawImg(self):
self.readEntrys()
self.fig.clf()
if 'z' in self.data:
ax = self.fig.add_subplot(projection='3d')
ax.plot(self.data['x'], self.data['y'], self.data['z'])
else:
ax = self.fig.add_subplot()
ax.plot(self.data['x'], self.data['y'])
self.fig.subplots_adjust(left=0.1, right=0.95, top=0.95, bottom=0.08)
self.canvas.draw()
源代碼
最后，將所有代碼附在后面
import tkinter as tk
import tkinter.ttk as ttk
import matplotlib as mpl
mpl.use('TkAgg')
from matplotlib.backends.backend_tkagg import (
FigureCanvasTkAgg, NavigationToolbar2Tk)
from matplotlib.figure import Figure
import numpy as np
def setEntry(e, text):
e.delete(0, "end")
e.insert(0, text)
def detectArray(s):
return s.rstrip('0123456789:, ')==''
class DarwSystem():
def __init__(self):
self.root = tk.Tk()
self.root.title("數(shù)據(jù)展示工具")
self.data = {}
self.setFrmCtrl()
self.setFrmFig()
self.root.mainloop()
def setFrmCtrl(self):
frmCtrl = ttk.Frame(self.root,width=320)
frmCtrl.pack(side=tk.RIGHT, fill=tk.Y)
frm = ttk.Frame(frmCtrl, width=320)
frm.pack(side=tk.TOP, fill=tk.X)
ttk.Button(frm, text="繪圖",width=5,
command=self.btnDrawImg).pack(side=tk.LEFT)
self.entrys = {}
for flag in 'xyz':
frm = ttk.Frame(frmCtrl)
frm.pack(side=tk.TOP, fill=tk.X)
self.setFrmAxis(frm, flag)
def setFrmAxis(self, frm, flag):
tk.Label(frm, text=flag).pack(side=tk.LEFT)
self.entrys[flag] = tk.Entry(frm)
self.entrys[flag].pack(side=tk.LEFT, fill=tk.X)
def readEntrys(self):
self.data = {}
for flag in 'xyz':
label = self.entrys[flag].get()
if label=="": continue
self.data[flag] = eval(self.entrys[flag].get())
if flag =='x' : x = self.data['x']
elif flag =='y' : y = self.data['y']
def btnDrawImg(self):
self.readEntrys()
self.fig.clf()
if 'z' in self.data:
ax = self.fig.add_subplot(projection='3d')
ax.plot(self.data['x'], self.data['y'], self.data['z'])
else:
ax = self.fig.add_subplot()
ax.plot(self.data['x'], self.data['y'])
self.fig.subplots_adjust(left=0.1, right=0.95, top=0.95, bottom=0.08)
self.canvas.draw()
def setFrmFig(self):
frmFig = ttk.Frame(self.root)
frmFig.pack(side=tk.LEFT,fill=tk.BOTH,expand=tk.YES)
self.fig = Figure()
self.canvas = FigureCanvasTkAgg(self.fig,frmFig)
self.canvas.get_tk_widget().pack(
side=tk.TOP,fill=tk.BOTH,expand=tk.YES)
self.toolbar = NavigationToolbar2Tk(self.canvas,frmFig,
pack_toolbar=False)
self.toolbar.update()
self.toolbar.pack(side=tk.RIGHT)
if __name__ == "__main__":
test = DarwSystem()
以上就是使用Python實現(xiàn)從零開始打造一個三維繪圖系統(tǒng)的詳細內(nèi)容，更多關(guān)于Python三維繪圖系統(tǒng)的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
python中requests庫session對象的妙用詳解
這篇文章主要介紹了python中requests庫session對象的妙用詳解，具有一定參考價值，需要的朋友可以了解下。2017-10-10
Python OpenCV 調(diào)用攝像頭并截圖保存功能的實現(xiàn)代碼
這篇文章主要介紹了Python OpenCV 調(diào)用攝像頭并截圖保存功能，本文通過兩段實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2019-07-07
大家好，本篇文章主要講的是Python的輸出格式化和進制轉(zhuǎn)換介紹，感興趣的同學趕快來看一看吧，對你有幫助的話記得收藏一下2022-01-01
Python利用for循環(huán)修改列表的常見錯誤解析
在Python編程中,for循環(huán)是最常用的控制結(jié)構(gòu)之一,而列表（list）是最基礎的數(shù)據(jù)結(jié)構(gòu)之一,但在使用for循環(huán)遍歷列表并對列表進行增刪改操作時,往往會遇到意想不到的錯誤,下面我們就來深入研究一下吧2025-06-06
這篇文章主要介紹了python中hashlib模塊用法示例，具有一定參考價值，需要的朋友可以了解下。2017-10-10
這篇文章主要為大家詳細介紹了python實現(xiàn)簡易計算器功能，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-02-02
在本篇文章里小編給大家整理了關(guān)于python中數(shù)字是否為可變類型的相關(guān)文章內(nèi)容，需要的朋友們可以學習下。2020-07-07
python/golang實現(xiàn)循環(huán)鏈表的示例代碼
這篇文章主要介紹了python/golang如何實現(xiàn)循環(huán)鏈表，幫助大家更好的理解和學習循環(huán)鏈表的實現(xiàn)方法，感興趣的朋友可以了解下2020-09-09
Python JSON格式數(shù)據(jù)的提取和保存的實現(xiàn)
這篇文章主要介紹了Python JSON格式數(shù)據(jù)的提取和保存的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2019-03-03
淺談關(guān)于Python3中venv虛擬環(huán)境
這篇文章主要介紹了淺談關(guān)于Python3中venv虛擬環(huán)境，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-08-08