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, 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__' :

#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 = 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()