matplotlib API 入门
matplotlib API 函数(如plot和close)都位于matplotlib.pyplot模块中,通常的引入方式如下:
import matplotlib.pyplot as plt
Figure和Subplot
matplotlib的图像都位于Figure对象中。可以用plt.figure创建一个新的Figure.
fig=plt.figure()
这时会弹出一个空窗口。当Spyder不能弹出一个空窗口时,选择Spyder的tools下拉菜单preferences,选择IPython console窗口中Graphics,将Backend从Inline改成Qt5,注意:应用后,要重启Spyder
plt.figure有一些选项,特别是figsize,它用于确保当图片保存到磁盘时具有一定的大小和纵横比。
不能通过空Figure绘图。必须用add_subplot创建一个(add_subplot(1,1,1))或多个subplot才行。
ax1=fig.add_subplot(2,2,1)#图像是2x2的,当前选中的是4个subplot中的第一个(编号从1开始)
ax2=fig.add_subplot(2,2,2)
ax3=fig.add_subplot(2,2,3)
#这段代码需要在fig=plt.figure()的基础上绘制图像,也就是说不要把fig=plt.figure()创建的空窗口关闭
如果这时发出一条绘图命令(如plt.plot([1.5,3.5,-2,1.6])),matplotlib就会在最后一个用过的subplot上进行绘制。
from numpy.random import randn
plt.plot(randn(50).cumsum(),'k--')#“k--”是一个线型选项,用于告诉matplotlib绘制黑色虚线图
#为当前图像或最后一个使用的subplot对象绘制,所以图出现在第3个subplot中
上面那些由fig.add_subplot所返回的对象是AxesSubplot对象
_=ax1.hist(randn(100),bins=20,color='k',alpha=0.3)
ax2.scatter(np.arange(30),np.arange(30)+3*randn(30))#指定某个图像对象,进行绘制
plt.subplot方法可以创建一个新的Figure,并返回一个含有已创建的subplot对象的NumPy数组
fig,axes=plt.subplots(2,3)
In [6]: axes #这个和书上的返回结果不同
Out[6]:
array([[<matplotlib.axes._subplots.AxesSubplot object at 0x000000000E8CB9B0>,
<matplotlib.axes._subplots.AxesSubplot object at 0x000000000D73FD30>,
<matplotlib.axes._subplots.AxesSubplot object at 0x000000000D772E80>],
[<matplotlib.axes._subplots.AxesSubplot object at 0x000000000D7AAFD0>,
<matplotlib.axes._subplots.AxesSubplot object at 0x000000000D7EE048>,
<matplotlib.axes._subplots.AxesSubplot object at 0x000000000E91CF60>]], dtype=object)
可以像对二维数组一样,对axes数组进行索引,例如axes[0,1],代表第1行第2个axes对象.可以通过sharex和sharey指定subplot应该具有相同的X轴或Y轴。
调整subplot周围的间距
默认情况下,matplotlib会在subplot外围留下一定的边距,并在subplot之间留下一定的间距。间距和图像的高度和宽度有关。利用Figure的subplots_adjust方法可以修改间距。
subplots_adjust(left=None,bottom=None,right=None,wspace=None,hspace=None)
wspace和hspace用于控制宽度和高度的百分比,可以用作subplot之间的间距。
fig,axes=plt.subplots(2,2,sharex=True,sharey=True)#产生两行两列共4个空白子图
for i in range(2):
for j in range(2):
axes[i,j].hist(randn(500),bins=50,color='k',alpha=0.5)
plt.subplots_adjust(wspace=0,hspace=0)#各subplot之间没有间距,且标签重叠了
颜色、标记和线型
matplotlib的plot函数接受一组X和Y坐标,还可以接受一个表示颜色和线型的字符串缩写。例如,要根据x和y绘制绿色虚线。
ax.plot(x,y,'g--') #这里的ax表示某个图像对象,比如上文的as1,as2,as3
通过下面这种更为明确的方式也能得到同样的效果:
ax.plot(x,y,linestyle='--',color='g') #linestyle用于指明线型
常用的颜色都有一个缩写词,要