# CSV文件數據圖形化工具 相關文檔： [_設計自己的Trait編輯器_](traitsui_manual_custom_editor.html) 采用 [_在traitsUI中使用的matplotlib控件_](example_mpl_figure_editor.html) 制作的CSV文件數據繪圖工具。   ``` # -*- coding: utf-8 -*- from matplotlib.figure import Figure from mpl_figure_editor import MPLFigureEditor from enthought.traits.ui.api import * from enthought.traits.api import * import csv class DataSource(HasTraits): """ 數據源，data是一個字典，將字符串映射到列表 names是data中的所有字符串的列表 """ data = DictStrAny names = List(Str) def load_csv(self, filename): """ 從CSV文件讀入數據，更新data和names屬性 """ f = file(filename) reader = csv.DictReader(f) self.names = reader.fieldnames for field in reader.fieldnames: self.data[field] = [] for line in reader: for k, v in line.iteritems(): self.data[k].append(float(v)) f.close() class Graph(HasTraits): """ 繪圖組件，包括左邊的數據選擇控件和右邊的繪圖控件 """ name = Str # 繪圖名，顯示在標簽頁標題和繪圖標題中 data_source = Instance(DataSource) # 保存數據的數據源 figure = Instance(Figure) # 控制繪圖控件的Figure對象 selected_xaxis = Str # X軸所用的數據名 selected_items = List # Y軸所用的數據列表 clear_button = Button(u"清除") # 快速清除Y軸的所有選擇的數據 view = View( HSplit( # HSplit分為左右兩個區域，中間有可調節寬度比例的調節手柄 # 左邊為一個組 VGroup( Item("name"), # 繪圖名編輯框 Item("clear_button"), # 清除按鈕 Heading(u"X軸數據"), # 靜態文本 # X軸選擇器，用EnumEditor編輯器，即ComboBox控件，控件中的候選數據從 # data_source的names屬性得到 Item("selected_xaxis", editor= EnumEditor(name="object.data_source.names", format_str=u"%s")), Heading(u"Y軸數據"), # 靜態文本 # Y軸選擇器，由于Y軸可以多選，因此用CheckBox列表編輯，按兩列顯示 Item("selected_items", style="custom", editor=CheckListEditor(name="object.data_source.names", cols=2, format_str=u"%s")), show_border = True, # 顯示組的邊框 scrollable = True, # 組中的控件過多時，采用滾動條 show_labels = False # 組中的所有控件都不顯示標簽 ), # 右邊繪圖控件 Item("figure", editor=MPLFigureEditor(), show_label=False, width=600) ) ) def _name_changed(self): """ 當繪圖名發生變化時，更新繪圖的標題 """ axe = self.figure.axes[0] axe.set_title(self.name) self.figure.canvas.draw() def _clear_button_fired(self): """ 清除按鈕的事件處理 """ self.selected_items = [] self.update() def _figure_default(self): """ figure屬性的缺省值，直接創建一個Figure對象 """ figure = Figure() figure.add_axes([0.05, 0.1, 0.9, 0.85]) #添加繪圖區域，四周留有邊距 return figure def _selected_items_changed(self): """ Y軸數據選擇更新 """ self.update() def _selected_xaxis_changed(self): """ X軸數據選擇更新 """ self.update() def update(self): """ 重新繪制所有的曲線 """ axe = self.figure.axes[0] axe.clear() try: xdata = self.data_source.data[self.selected_xaxis] except: return for field in self.selected_items: axe.plot(xdata, self.data_source.data[field], label=field) axe.set_xlabel(self.selected_xaxis) axe.set_title(self.name) axe.legend() self.figure.canvas.draw() class CSVGrapher(HasTraits): """ 主界面包括繪圖列表，數據源，文件選擇器和添加繪圖按鈕 """ graph_list = List(Instance(Graph)) # 繪圖列表 data_source = Instance(DataSource) # 數據源 csv_file_name = File(filter=[u"*.csv"]) # 文件選擇 add_graph_button = Button(u"添加繪圖") # 添加繪圖按鈕 view = View( # 整個窗口分為上下兩個部分 VGroup( # 上部分橫向放置控件，因此用HGroup HGroup( # 文件選擇控件 Item("csv_file_name", label=u"選擇CSV文件", width=400), # 添加繪圖按鈕 Item("add_graph_button", show_label=False) ), # 下部分是繪圖列表，采用ListEditor編輯器顯示 Item("graph_list", style="custom", show_label=False, editor=ListEditor( use_notebook=True, # 是用多標簽頁格式顯示 deletable=True, # 可以刪除標簽頁 dock_style="tab", # 標簽dock樣式 page_name=".name") # 標題頁的文本使用Graph對象的name屬性 ) ), resizable = True, height = 0.8, width = 0.8, title = u"CSV數據繪圖器" ) def _csv_file_name_changed(self): """ 打開新文件時的處理，根據文件創建一個DataSource """ self.data_source = DataSource() self.data_source.load_csv(self.csv_file_name) del self.graph_list[:] def _add_graph_button_changed(self): """ 添加繪圖按鈕的事件處理 """ if self.data_source != None: self.graph_list.append( Graph(data_source = self.data_source) ) if __name__ == "__main__": csv_grapher = CSVGrapher() csv_grapher.configure_traits() ```