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.figurefigsize = self.figsize)
    if self.axis != None :
      pylab.axisself.axis)
    pylab.titleself.title)
    pylab.ylabelself.ylabel)
    ax = pylab.gca)
    pylab.gridself.grid)
    if self.xaxis_locator != None :
      ax.xaxis.set_major_locator pylab.MultipleLocatorself.xaxis_locator) )
    if self.yaxis_locator != None :
      ax.yaxis.set_major_locator pylab.MultipleLocatorself.yaxis_locator) )
    self.lineList = []
    self.id = 1
  def addlineself, lineConf) :
    self.lineList.appendself.id, lineConf))
    self.id += 1
    return {'id' : self.id - 1}
  def removelineself, lineId) :
    for i in rangelenself.lineList)) :
      id, conf = self.lineList[i]
      if id == lineId :
        del self.lineList[i]
        break
    else :
      return {'status' : -1}
    print lenself.lineList)
    return {'status' : 0}
  def __parselineConfself, lineConf) :
    X = lineConf['X']
    Y = lineConf['Y']
    marker = lineConf.get'marker',None)
    color = lineConf.get'color', random.choiceMiniPlotTool.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 plotSingleLineself, lineConf):
    X, Y, marker, color, markerfacecolor, label, linewidth, linestyle = self.__parselineConflineConf)
    pylab.plotX, Y, marker = marker, color = color, markerfacecolor = markerfacecolor, label=label, linewidth = linewidth, linestyle = linestyle)
    pylab.legendloc = self.legend_loc)
  def plotself) :
    colors = [MiniPlotTool.basecolors[i % lenMiniPlotTool.basecolors)] for i in rangelenself.lineList))]
    for i in rangelenself.lineList)) :
      id, conf = self.lineList[i]
      if conf.get'color',None) :
        conf['color'] = colors[i]
      X, Y, marker, color, markerfacecolor, label, linewidth, linestyle = self.__parselineConfconf)
      pylab.plotX, Y, marker = marker, color = color, markerfacecolor = markerfacecolor, label=label, linewidth = linewidth, linestyle = linestyle)
    pylab.legendloc = self.legend_loc)
  def showself) :
    pylab.show)
if __name__ == '__main__' :
  #test
  baseConfig = {
    #'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 = MiniPlotToolbaseConfig)
  X = [ i for i in range10)]
  Y = [random.randint1,10) for i in range10)]
  Y2 = [random.randint1,10) for i in range10)]
  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.plotSingleLinelineConf)
  print tool.addlinelineConf)
  print tool.addlinelineConf2)
  #print tool.removeline1)
  tool.plot)
  tool.show)