一文帶你學會如何利用Python實現一個三維繪圖系統(tǒng)

更新時間：2023年09月06日 10:30:41 作者：微小冷

tkinter是Python標準庫中自帶的GUI工具,使用十分方便,所以本文旨在帶大家學會如何將matplotlib嵌入到tkinter中并繪制三維繪圖系統(tǒng),感興趣的可以了解下

將圖像嵌入tkinter

tkinter是Python標準庫中自帶的GUI工具，使用十分方便，如能將matplotlib嵌入到tkinter中，就可以做出相對專業(yè)的數據展示系統(tǒng)，很有競爭力。

在具體實現之前，可以先看一下典型的 matplotlib 窗口

import numpy as np
import matplotlib.pyplot as plt
plt.plot(np.arange(100))
plt.show()

這個圖由兩部分構成，分別是上面用于繪圖的 FigureCanvasTkAgg 畫布，以及下方的工具欄 NavigationToolbar2Tk ，這兩個模塊在地位上和tkinter中的組件是等同的。

除此之外，繪圖窗口Figure也是一個獨立部件，故而將matplotlib嵌入到tkinter中，最少需要用到下面這些模塊

import tkinter as tk
import tkinter.ttk as ttk
import matplotlib as mpl
mpl.use('TkAgg')        # 啟用tkinter渲染matplotlib，從而可以嵌入到tkinter中
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import (
    FigureCanvasTkAgg, NavigationToolbar2Tk)
from matplotlib.figure import Figure

接下來可以創(chuàng)建一個窗口，來演示一下matplotlib嵌入tkinter中的過程，首先創(chuàng)建一個窗口，并為其添加兩個Frame，右邊用于添加組件，左邊用于嵌入圖像，代碼如下

root = tk.Tk()
root.title("數據展示工具")
frmCtrl = ttk.Frame(root, width=200)
frmCtrl.pack(side=tk.RIGHT)
frmFigure = ttk.Frame(root)
frmFigure.pack(side=tk.LEFT, fill=tk.BOTH, expand=tk.YES)

嵌入圖像的部分又分為畫布和工具欄，為了便于演示，下面先在繪圖窗口中畫一條直線，并將其導入畫布，然后將畫布放置到frmFigure上。至于工具欄，在關聯畫布和Frame之后需要更新一下，整體代碼如下

fig = Figure()
ax = fig.add_subplot()
ax.plot(np.arange(100))
canvas = FigureCanvasTkAgg(fig,frmFigure)
canvas.get_tk_widget().pack(
    side=tk.TOP,fill=tk.BOTH,expand=tk.YES)
toolbar = NavigationToolbar2Tk(canvas,frmFigure,
    pack_toolbar=False)
toolbar.update()
toolbar.pack(side=tk.RIGHT)

至此，就已經完成了圖像的嵌入工作，最后調用mainloop，讓窗口一直顯示

root.mainloop()

結果如下

簡單的繪圖系統(tǒng)

在理解matplotlib嵌入到tkinter中的原理之后，就已經具備了打造繪圖系統(tǒng)的技術基礎，接下來要做的，就是做一個較有可讀性的繪圖類，其實就是把前面的代碼封裝到class里而已，代碼如下

import tkinter as tk
import tkinter.ttk as ttk
import matplotlib as mpl
mpl.use('TkAgg')
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import (
    FigureCanvasTkAgg, NavigationToolbar2Tk)
from matplotlib.figure import Figure
class DarwSystem():
    def __init__(self):
        self.root = tk.Tk()
        self.root.title("數據展示工具")
        frmCtrl = ttk.Frame(self.root,width=320)
        frmCtrl.pack(side=tk.RIGHT, fill=tk.Y)
        self.setFrmCtrl(frmCtrl)
        frmFig = ttk.Frame(self.root)
        frmFig.pack(side=tk.LEFT,fill=tk.BOTH,expand=tk.YES)
        self.setFrmFig(frmFig)
        self.root.mainloop()
    def setFrmCtrl(self, frmCtrl):
        pass
    def setFrmFig(self, frmFig):
        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)

其中，setFrmCtrl用于設置控制面板，后續(xù)會實現諸多功能；setFrmFig用于設置繪圖界面，其中self.fig就是繪圖窗口，后續(xù)若要畫圖，都要在這里設置坐標軸。

最簡單的繪圖系統(tǒng)，也至少需要三個部件，分別用于輸入x值、y值以及點擊繪圖按鈕，由于x,y都是繪圖坐標，在控件形式上也高度雷同，從而setFrmCtrl函數可以先寫為下面的形式

def setFrmCtrl(self, frmCtrl):
    frm = ttk.Frame(frmCtrl)
    frm.pack(side=tk.TOP, fill=tk.X)
    self.setCtrlButtons(frm)
    self.data = {}
    self.entrys = {}
    for flag in 'xy':
        frm = ttk.Frame(frmCtrl)
        frm.pack(side=tk.TOP, fill=tk.X)
        self.setFrmAxis(frm, flag)

第一個frm用于存放控制按鈕；self.data用于存放坐標值，self.entrys中用于存放坐標的輸入框，后面的循環(huán)為具體的布局過程。

setCtrlButtons 是具體的控制按鈕的布局函數，而 setFrmAxis 為坐標軸的布局函數，定義如下

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 setCtrlButtons(self, frm):
    tk.Button(frm, text="繪圖",width=5,
        command=self.btnDrawImg).pack(side=tk.LEFT)
# 繪圖函數
def btnDrawImg(self):
    pass

其中btnDrawImg是繪圖函數，簡單起見，這里用eval函數直接讀取python表達式，同時為了讓不熟悉Python的人也可以順利生成x序列，將np.linspace隱去。則x和y的讀取過程可寫為

def btnDrawImg(self):
    x = eval(f"np.linspace({self.entrys['x'].get()})")
    self.data['y'] = eval(self.entrys['y'].get())
    self.data['x'] = x
    self.drawPlot()

self.drawPlot就是核心的繪圖函數，主要流程與命令行調用plt如出一轍，首先創(chuàng)建一個坐標軸，然后在坐標軸上繪圖，區(qū)別是最后需要調用self.canvas中的引擎來完成圖像繪制

def drawPlot(self):
    self.fig.clf()
    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()

結果如下

導入數據

單純從作圖的角度來說，更多情況是已經有了一組數據，然后需要將其繪制。這組數據可能是txt格式的，也可能是csv格式的，還可能是二進制數據。當然，這些一會兒在想，首先就是要添加一個按鈕，將 setCtrlButtons 函數添加一行：

def setCtrlButtons(self, frm):
    ttk.Button(frm, text="繪圖",width=5,
        command=self.btnDrawImg).pack(side=tk.LEFT)
    ttk.Button(frm, text="加載",width=5,
        command=self.btnLoadData).pack(side=tk.LEFT)

然后就可以考慮 self.btnLoadData 函數了。簡潔起見，以后將不再具體展示 setCtrlButtons 的具體代碼，而只是寫出新增的代碼。

加載數據，其實就是加載文件，那么就需要用到文件對話框

from tkinter.filedialog import askopenfile

self.btnLoadData 函數，如果只是想實現一個最簡單的功能，那么可以寫為

def btnLoadData(self):
    name = askopenfilename()
    data = np.genfromtxt(name)
    if data.shape[1] < 2:
        return
    self.data['x'] = data[:,0]
    self.data['y'] = data[:,1]
    self.drawPlot()

效果如下

現在，我們有了兩種數據生成模式，一是用語法生成，二是通過加載得到。但目前二者并不兼容。為此，可以為x和y的輸入框添加一個標識，比如當x或者y的輸入框中是 data 的時候，再點擊繪圖，就可以選中加載后的數據。

由于tkinter中輸入Entry內容比較繁瑣，所以封裝一個全局的函數專門用于更改Entry內容

def setEntry(e, text):
    e.delete(0, "end")
    e.insert(0, text)

接下來，將加載數據函數和繪圖函數分別改寫為

def btnLoadData(self):
    name = askopenfilename()
    data = np.genfromtxt(name)
    if data.shape[1] < 2:
        return
    for i,key in enumerate('xy'):
        self.data[key] = data[:,i]
        setEntry(self.entrys[key], 'data')
def btnDrawImg(self):
    xLab = self.entrys['x'].get()
    if xLab != "data":
        x = eval(f"np.linspace({xLab})")
        self.data['x'] = x
    else:
        x = self.data['x']
    yLab = self.entrys['y'].get()
    if yLab != "data":
        self.data['y'] = eval(yLab)
    self.drawPlot()

在btnLoadData函數中，取消了繪圖功能，而是在導入數據后，將x和y的輸入框設置為"data"。

而繪圖函數中，檢測x和y輸入框的內容，如果是data，那么說明已經讀取到了相關數據，就直接調用，而非重新生成。

效果如下

三維繪圖

三維繪圖需要一個新的坐標變z，由于此前在設置x和y的時候，用到了self.entrys來存放坐標，所以更改起來十分便捷，只需在setFrmCtrl的循環(huán)中加一個’z’就可以了

def setFrmCtrl(self, frmCtrl):
    # 前面不用管，后面也不用管，只要把in 'xy'改為 'xyz'
    for flag in 'xyz':
        # 里面也不用管

相應地，加載數據的函數也略作修改

def btnLoadData(self):
    # 前面不用動，后面也不用動，只要把'xy'換成'xyz'
    for i,key in enumerate('xyz'):
        # 后面不用動

但隨著z軸的出現，y軸也有可能是自變量，考慮到x和y都有可能用類似 1,1,5 的形式生成，所以先做一個檢測數組的全局函數

def detectArray(s):
    return s.rstrip('0123456789:, ')==''

然后新建readEntrys函數以讀取輸入框，考慮到在函數表達式中，用x和y指代self.data[‘x’]和selfdata[‘y’]，所以需要新建局部變量x和y，以確保eval函數的正常使用。

def readEntrys(self):
    for flag in 'xyz':
        label = self.entrys[flag].get()
        if label=="":
                continue
        if label != 'data':
            if detectArray(label):
                label = f"np.linspace({label})"
            self.data[flag] = eval(self.entrys[flag].get())
        if flag =='x' : x = self.data['x']
        elif flag =='y' : y = self.data['y']        
    if self.entrys['z'].get()=="":
        del self.data['z']

最后，就是繪圖功能的實現，由于有了readEntrys函數，從而btnDrawImg函數變得更加專注，只需復制調用專門的繪圖函數就可以了。三維繪圖函數和二維繪圖函數其實沒什么區(qū)別，只要繪制的還是plot圖，區(qū)別只是多加了一個z軸坐標而已。

def btnDrawImg(self):
    self.readEntrys()
    self.fig.clf()
    if 'z' in self.data: 
        self.drawPlot3D()
    else:
        self.drawPlot()
    self.fig.subplots_adjust(left=0.1, right=0.95, top=0.95, bottom=0.08)
    self.canvas.draw()
def drawPlot(self):
    self.fig.clf()
    ax = self.fig.add_subplot()
    ax.plot(self.data['x'], self.data['y'])
def drawPlot3D(self):
    ax = self.fig.add_subplot(projection='3d')
    ax.plot(self.data['x'], self.data['y'], self.data['z'])

效果如下

源代碼

本文是在這四篇博客的基礎之上，做適當精簡而總結的：將matplotlib嵌入到tkinter |簡單的繪圖系統(tǒng)|數據導入|三維繪圖系統(tǒng)

其基本思路是，以第四篇博客的源代碼為基準，優(yōu)化這個代碼的設計過程，使之更易于學習，所以源代碼并沒有發(fā)生變化。

import tkinter as tk
import tkinter.ttk as ttk
from tkinter.filedialog import askopenfilename
import matplotlib as mpl
mpl.use('TkAgg')
import matplotlib.pyplot as plt
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("數據展示工具")
        self.data = {}
        frmCtrl = ttk.Frame(self.root,width=320)
        frmCtrl.pack(side=tk.RIGHT, fill=tk.Y)
        self.setFrmCtrl(frmCtrl)
        frmFig = ttk.Frame(self.root)
        frmFig.pack(side=tk.LEFT,fill=tk.BOTH,expand=tk.YES)
        self.setFrmFig(frmFig)
        self.root.mainloop()
    def setFrmCtrl(self, frmCtrl):
        frm = ttk.Frame(frmCtrl, width=320)
        frm.pack(side=tk.TOP, fill=tk.X)
        self.setCtrlButtons(frm)
        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 setCtrlButtons(self, frm):
        ttk.Button(frm, text="繪圖",width=5,
            command=self.btnDrawImg).pack(side=tk.LEFT)
        ttk.Button(frm, text="加載",width=5,
            command=self.btnLoadData).pack(side=tk.LEFT)
    def btnLoadData(self):
        name = askopenfilename()
        data = np.genfromtxt(name)
        for i, flag in enumerate('xyz'):
            if i >= data.shape[1]:
                return
            self.data[flag] = data[:,i]
            setEntry(self.entrys[flag], 'data')
    def readEntrys(self):
        for flag in 'xyz':
            label = self.entrys[flag].get()
            if label=="":
                continue
            if label != 'data':
                if detectArray(label):
                    label = f"np.linspace({label})"
                self.data[flag] = eval(self.entrys[flag].get())
            if flag =='x' : x = self.data['x']
            elif flag =='y' : y = self.data['y']        
        if self.entrys['z'].get()=="":
            del self.data['z']
    def btnDrawImg(self):
        self.readEntrys()
        self.fig.clf()
        if 'z' in self.data:
            self.drawPlot3D()
        else:
            self.drawPlot()
        self.fig.subplots_adjust(left=0.1, right=0.95, top=0.95, bottom=0.08)
        self.canvas.draw()
    def drawPlot3D(self):
        ax = self.fig.add_subplot(projection='3d')
        ax.plot(self.data['x'], self.data['y'], self.data['z'])
    def drawPlot(self):
        ax = self.fig.add_subplot()
        ax.plot(self.data['x'], self.data['y'])
    def setFrmFig(self, frmFig):
        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實現一個三維繪圖系統(tǒng)的詳細內容，更多關于Python三維繪圖的資料請關注腳本之家其它相關文章！

您可能感興趣的文章:

淺談python numpy中nonzero()的用法
下面小編就為大家分享一篇淺談python numpy中nonzero()的用法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-04-04
Python中生成Epoch的方法
下面小編就為大家?guī)硪黄狿ython中生成Epoch的方法。小編覺得挺不錯的，現在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2017-04-04
使用Python中OpenCV和深度學習進行全面嵌套邊緣檢測
這篇文章主要介紹了使用Python中OpenCV和深度學習進行全面嵌套邊緣檢測，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2021-05-05
pandas如何篩選某個列值是否位于某個列表內
這篇文章主要介紹了pandas如何篩選某個列值是否位于某個列表內問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2024-02-02
Python實現的KMeans聚類算法實例分析
這篇文章主要介紹了Python實現的KMeans聚類算法,結合實例形式較為詳細的分析了KMeans聚類算法概念、原理、定義及使用相關操作技巧,需要的朋友可以參考下
2018-12-12
一文帶你掌握Python中textwrap庫文本包裝的藝術
在Python編程中,處理文本是一項基礎且常見的任務,textwrap模塊正是為此而生,它提供了一系列簡單而強大的工具,幫助我們優(yōu)雅地完成文本包裝和格式化工作,下面就跟隨小編來看看它的具體使用吧
2024-12-12
Python對于json數據鍵值對遍歷
這篇文章主要介紹了一文搞定Python中對于json數據鍵值對遍歷,Python中可以通過字典或者列表的遍歷方式來遍歷JSON格式的數據中的鍵值對，具體的遍歷方法取決于你將JSON數據解析后得到了一個字典對象還是一個列表對象，以及JSON數據的結構，需要的朋友可以參考下
2023-04-04
對python特殊函數 __call__()的使用詳解
今天小編就為大家分享一篇對python特殊函數 __call__()的使用詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-07-07
python實現調用其他python腳本的方法
python實現調用其他python腳本的方法,是一個比較實用的技巧,需要的朋友可以參考下
2014-10-10
python讀寫csv文件的方法
這篇文章主要介紹了python讀寫csv文件的方法，本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值 ,需要的朋友可以參考下
2019-08-08