matplotlib 是python最著名的绘图库,它提供了一整套和matlab相似的命令API,十分适合交互式地进行制图。该包下面有很多对象,比如pylab,pyplot等等,pylab集成了pyplot和numpy两个模块,能够进行快速绘图。Pylab和pyplot(http://matplotlib.org/api/pyplot_api.html)都能够通过对象或者属性对图像进行操作。Pyplot下也有很多对象,如figure,Axes对象等等,对图像进行细节处理。下面通过一些简单和常用的例子进行操作。
- 通过pyplot画散点图,调用scatter函数,同时该函数也可以画不同大小点的散点图。例子如下:
# .*.coding : utf-8 .*.
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0,20,2)
y = np.linspace(0,20,10)
plt.figure()
plt.scatter(x,y,c='r',marker='*')
plt.xlabel('X')
plt.ylabel('Y')
plt.title('$X*Y$')
plt.show()
2.画曲线图,通过plot函数画曲线图,如下:
import numpy as np import random from matplotlib import pylab as pl from matplotlib import pyplot as plt #画一条曲线图 plt.figure(figsize=(8,6)) #设置图形的尺寸,单位为英尺 x = np.random.randn(10)*10 y = np.pi*np.sin(x) + 8 x1 = np.arange(0,10,1) #lines = plt.plot(x,y) plt.plot(x,y,label = '$np.pi*np.sin(x)+8$',color='red',linewidth = 2)plt.plot(x,y,'bo') plt.title('Y=a*sin(x)+b') plt.ylabel('Function-Y') plt.xlabel('Var-X') plt.legend() #图形的右上角显示图形的标签,和上面的plt.plo(x,y,label='')有关 #plt.setp(lines,color = 'r',linewidth = 2.5) plt.show()
3. 将图像画在多个轴上面,并在一个figure中。可以通过subplot,还有ubplot2grid等等方式均可实现。
import numpy as np
import matplotlib.pylab as plt
'''将图形画在一个矩阵制定的表中,利用subplot,plot()中放有多个自变量和函数,画多坐标图'''
def f(t):
return np.exp(-t) * np.cos(2*np.pi*t)
t1 = np.arange(0.0, 5.0, 0.1)
t2 = np.arange(0.0, 5.0, 0.02)
plt.figure(1)
plt.subplot(211) #subplot(numRows, numCols, plotNum),如:subplot(221)
plt.plot(t1, f(t1), 'bo', t2, f(t2), 'k')
plt.title('pylab first example',fontsize = 16)
plt.subplot(212)
plt.plot(t2, np.cos(2*np.pi*t2), 'r--')
plt.title('pylab second example',fontsize = 16)
plt.show()
4. 使用text将文本表现在图中,并对图像之间进行设置,坐标设置等。
# -*- coding: utf-8 -*- import matplotlib.pyplot as plt #通过对象的形式文本作图 fig = plt.figure() fig.suptitle('bold figure suptitle', fontsize=14, fontweight='bold')#figure的标题 ax = fig.add_subplot(111) #figure下的坐标图ax fig.subplots_adjust(top=0.85) ax.set_title('axes title') #给ax设置标题 ax.set_xlabel('xlabel') #给ax设置横坐标 ax.set_ylabel('ylabel') #给ax设置纵坐标 ax.text(3, 8, 'boxed italics text in data coords', style='italic', bbox={'facecolor':'red', 'alpha':0.5, 'pad':10}) ax.text(2, 6, r'an equation: $E=mc^2$', fontsize=15) ax.text(3, 2, unicode('unicode: Institut f\374r Festk\366rperphysik', 'latin-1')) ax.text(0.95, 0.01, 'colored text in axes coords', verticalalignment='bottom', horizontalalignment='right', transform=ax.transAxes, color='green', fontsize=15) ax.plot([2], [1], 'o') #ax画图,以及下面添加注释 ax.annotate('annotate', xy=(2, 1), xytext=(3, 4), arrowprops=dict(facecolor='black', shrink=0.05)) ax.axis([0, 10, 0, 10]) #设置横-纵轴的刻度 plt.show()
5. 通过hist画直方图
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0,1000,0.5)
y = np.random.rand(2000)*10
n,bins,patches = plt.hist(y,50,normed=1,alpha=0.8)
plt.title('Hist of Y')
plt.xlabel('Smarts')
plt.ylabel('NormProbability')
plt.text(2,0.12,'$\mu=10,\\sigma=20$')
plt.grid(True)
plt.show()
6. 通过bar画条形图
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0,1000,0.5)
y = np.random.rand(2000)*10
n,bins,patches = plt.hist(y,50,normed=1,alpha=0.8)
plt.title('Hist of Y')
plt.xlabel('Smarts')
plt.ylabel('NormProbability')
plt.text(2,0.12,'$\mu=10,\\sigma=20$')
plt.grid(True)
plt.show()
7. 对图形中的重要点进行注释,采用annotate实现改功能。具体用法可以采用官方文档。如下:
# .*. coding : utf-8 .*. import numpy as np import matplotlib.pyplot as plt x = np.arange(-10,10,1) y = 2*(x**2)-5*x+np.pi maxy = min(y) maxyx = x[np.argmin(y)] plt.figure() plt.plot(x,y,'-',color='red',linewidth=2.5,label='$2*(x^2)-5*x+np.pi$') plt.annotate('Max y at points',xy=(maxyx,maxy),xytext=(maxyx+1,maxy+10), arrowprops=dict(facecolor='black', shrink=0.01),fontsize=12) #xy为注释的点,xytext注释点文本的位置 plt.xlabel('X') plt.ylabel('Y') plt.xlim(-11,11) plt.title('plot of y=x^2') plt.show()
matplotlib官方文档中包含可很多画图的方法,如时间序列方面的处理,如:acorr;图像的处理,信号处理,数学公式等等。官方文档(http://matplotlib.org/py-modindex.html)