# -*- coding: utf-8 -*-
import pylab
import random
class MiniPlotTool :'''A mini tool to draw lines using pylab'''basecolors = ['red','green','yellow','blue','black','cyan','magenta']def __init__(self, baseConfig) :self.figsize = baseConfig.get('figsize',None)self.axis = baseConfig.get('axis',None)self.title = baseConfig.get('title','NoName')self.ylabel = baseConfig.get('ylabel','NoName')self.grid = baseConfig.get('grid',False)self.xaxis_locator = baseConfig.get('xaxis_locator',None)self.yaxis_locator = baseConfig.get('yaxis_locator',None)self.legend_loc = baseConfig.get('legend_loc',0)if self.figsize != None :pylab.figure(figsize = self.figsize)if self.axis != None :pylab.axis(self.axis)pylab.title(self.title)pylab.ylabel(self.ylabel)ax = pylab.gca()pylab.grid(self.grid)if self.xaxis_locator != None :ax.xaxis.set_major_locator( pylab.MultipleLocator(self.xaxis_locator) )if self.yaxis_locator != None :ax.yaxis.set_major_locator( pylab.MultipleLocator(self.yaxis_locator) )self.lineList = []self.id = 1def addline(self, lineConf) :self.lineList.append((self.id, lineConf))self.id += 1return {'id' : self.id - 1}def removeline(self, lineId) :for i in range(len(self.lineList)) :id, conf = self.lineList[i]if id == lineId :del self.lineList[i]breakelse :return {'status' : -1}print len(self.lineList)return {'status' : 0}def __parselineConf(self, lineConf) :X = lineConf['X']Y = lineConf['Y']marker = lineConf.get('marker',None)color = lineConf.get('color', random.choice(MiniPlotTool.basecolors))markerfacecolor = lineConf.get('markerfacecolor',color)label = lineConf.get('label','NoName')linewidth = lineConf.get('linewidth',1)linestyle = lineConf.get('linestyle','-')return X, Y, marker, color, markerfacecolor, label, linewidth, linestyledef plotSingleLine(self, lineConf):X, Y, marker, color, markerfacecolor, label, linewidth, linestyle = self.__parselineConf(lineConf)pylab.plot(X, Y, marker = marker, color = color, markerfacecolor = markerfacecolor, label=label, linewidth = linewidth, linestyle = linestyle)pylab.legend(loc = self.legend_loc)def plot(self) :colors = [MiniPlotTool.basecolors[i % len(MiniPlotTool.basecolors)] for i in range(len(self.lineList))]for i in range(len(self.lineList)) :id, conf = self.lineList[i]if conf.get('color',None) :conf['color'] = colors[i]X, Y, marker, color, markerfacecolor, label, linewidth, linestyle = self.__parselineConf(conf)pylab.plot(X, Y, marker = marker, color = color, markerfacecolor = markerfacecolor, label=label, linewidth = linewidth, linestyle = linestyle)pylab.legend(loc = self.legend_loc)def show(self) :pylab.show()
if __name__ == '__main__' :#testbaseConfig = {#'figsize' : (6,8),#'axis': [0,10,0,10],#'title' : 'hello title',#'ylabel' : 'hello ylabel','grid' : True,#'xaxis_locator' : 0.5,#'yaxis_locator' : 1,#'legend_loc' : 'upper right'}tool = MiniPlotTool(baseConfig)X = [ i for i in range(10)]Y = [random.randint(1,10) for i in range(10)]Y2 = [random.randint(1,10) for i in range(10)]lineConf = {'X' : X,'Y' : Y#'marker' : 'x',#'color' : 'b',#'markerfacecolor' : 'r',#'label' : '222',#'linewidth' : 3,#'linestyle' : '--'}lineConf2 = {'X' : X,'Y' : Y2,'marker' : 'o','color' : 'b','markerfacecolor' : 'r','label' : '222','linewidth' : 3,'linestyle' : '--'}#tool.plotSingleLine(lineConf)print tool.addline(lineConf)print tool.addline(lineConf2)#print tool.removeline(1)tool.plot()tool.show()