pylab 提供了比较强大的画图功能，但是函数和参数都比较多，很容易搞混。我们平常使用最多的应该是画线了。下面，简单的对一些常用的划线函数进行了封装，方便使用。
代码语言：javascript复制# -*- 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()运行效果图如下：
线属性:
颜色（color 简写为 c）：
蓝色： 'b' (blue) 绿色： 'g' (green) 红色： 'r' (red) 蓝绿色 (墨绿色)： 'c' (cyan) 红紫色 (洋红)： 'm' (magenta) 黄色： 'y' (yellow) 黑色： 'k' (black) 白色： 'w' (white) 灰度表示： e.g. 0.75 ([0,1] 内任意浮点数) RGB 表示法： e.g. '#2F4F4F' 或 (0.18, 0.31, 0.31) 任意合法的 html 中的颜色表示： e.g. 'red', 'darkslategray' 线型（linestyle 简写为 ls）：
实线： '-' 虚线： '--' 虚点线： '-.' 点线： ':' 点： '.' 点型（标记 marker）：
像素： ',' 圆形： 'o' 上三角： '^' 下三角： 'v' 左三角： '<' 右三角： '>'方形：'s'加号：' '叉形：'x'棱形：'D'细棱形：'d'三脚架朝下：'1'（就是丫） 三脚架朝上： '2' 三脚架朝左： '3' 三脚架朝右： '4' 六角形： 'h' 旋转六角形： 'H' 五角形： 'p' 垂直线： '|' 水平线： '_' gnuplot 中的 steps： 'steps' （只能用于 kwarg 中） 标记大小（markersize 简写为 ms）：
markersize： 实数 标记边缘宽度（markeredgewidth 简写为 mew）：
markeredgewidth：实数 标记边缘颜色（markeredgecolor 简写为 mec）：
markeredgecolor：颜色选项中的任意值 标记表面颜色（markerfacecolor 简写为 mfc）：
markerfacecolor：颜色选项中的任意值 透明度（alpha）：
alpha： [0,1] 之间的浮点数 线宽（linewidth）：
linewidth： 实数
