Python使用pylab库实现画线功能的方法详解

# -*- 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 = 1
  def addline(self,lineConf) :
    self.lineList.append((self.id,lineConf))
    self.id += 1
    return {'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]
        break
    else :
      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,linestyle
  def plotSingleLine(self,lineConf):
    X,linestyle = self.__parselineConf(lineConf)
    pylab.plot(X,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,linestyle = self.__parselineConf(conf)
      pylab.plot(X,linestyle = linestyle)
    pylab.legend(loc = self.legend_loc)
  def show(self) :
    pylab.show()
if __name__ == '__main__' :
  #test
  baseConfig = {
    #'figsize' : (6,8),#'axis': [0,10,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()