matplot API入门
如何引入:
import matplotlib.pyplot as plt
一个简单的例子:
import numpy as np
data = np.arange(10)
plt.plot(data)
print(data)
[0 1 2 3 4 5 6 7 8 9]
效果如下:
Figure和Subplot
matplot的图像都位于Figure对象中,相当于一个画板,如何创建Figure对象。
fig = plt.figure()
ax1 = fig.add_subplot(2, 2, 1)
ax2 = fig.add_subplot(2, 2, 2)
ax3 = fig.add_subplot(2, 2, 3)
想要实现绘图,必须先使用add_subplot
方法来创建画板。上面我们创建了三个2X2的画板:
plt.plot(np.random.randn(50).cumsum(), 'k--')
创建了一个随机曲线图:
最终matplotlib会选择在最后一张画板上绘图,如果没创建画板它会自动创建。
fig = plt.figure()
plt.plot(np.random.randn(50).cumsum(), 'k--')
其中k--
是一个线性选项,表示黑色虚线。
如果想对其中一个进行绘图,调用其实例的绘图方法即可:
_ = ax1.hist(np.random.randn(100), bins=20, color='k', alpha=0.3)
ax2.scatter(np.arange(30), np.arange(30) + 3 * np.random.randn(30))
ps: 其他图表类型可以查看文档
创建包含subplot网格的figure
通过plt.subplots
方法来创建一个包含多个subplot(图表)的Figure,并返回包含它们的NumPy数组
fig,axes = plt.subplots(2,3)
print(axes)
返回了一个包含subplot对象的2X3数组
[[<matplotlib.axes._subplots.AxesSubplot object at 0x11b6bca58>
<matplotlib.axes._subplots.AxesSubplot object at 0x11b6e3ac8>
<matplotlib.axes._subplots.AxesSubplot object at 0x11b70bcf8>]
[<matplotlib.axes._subplots.AxesSubplot object at 0x11b734ef0>
<matplotlib.axes._subplots.AxesSubplot object at 0x11b765048>
<matplotlib.axes._subplots.AxesSubplot object at 0x11b78d240>]]
这里我是这么理解的:
这是在一个figure中创建多个图表的一种方式。非常实用,因为可以轻松的对这个数组索引,可以像操作数据一样去操作这些对象。
当然多个图标进行对比首要条件就是刻度必须是相同的,pyplot.subplots
的sharex、sharey
属性就能做到这点。
pyplot.subplots的其他选项
调整subplots周围的间距
利用subplots_adjust方法来修改间距
表达式:
subplots_adjust(left=None, bottom=None, right=None, top=None, wspace=None, hspace=None)
wspace、hspace:代表宽度和高度的百分比,可以用来修改subplot之间的间距
看下例:
fig,axes = plt.subplots(2, 2, sharex=True, sharey=True)
for i in range(2):
for j in range(2):
axes[i,j].hist(np.random.randn(500), bins=50, color='k', alpha=0.5)
plt.subplots_adjust(wspace=0, hspace=0)
使用双重循环将4个subplot的间距都改为了0。
颜色、标记和线型
可以用matplotlib的plot函数来修改颜色、标记和线型。
表达式:
ax.plot(x, y, linesstyle='--', color='g')
其中x, y 表示X和Y坐标,linestyle修改线型,具体需要查看文档,color代表颜色,都有缩写具体查看文档
如果是线型图,可对数据加上锚点
from numpy.random import randn
plt.plot(randn(30).cumsum(), 'ko--')
也有另外一种写法,更加容易看懂些
plt.plot(randn(30), color='k', linestyle='dashed', marker='o')
在线性图中,除了锚点其他数据值都是以线性方式展现的。可以通过drawstyle选项做修改:
fig2 = plt.figure()
data = randn(30).cumsum()
ax1 = fig2.add_subplot(2, 2, 1)
ax2 = fig2.add_subplot(2, 2, 2)
ax1.plot(data, 'k--', label='Default')
ax2.plot(data, 'k--', drawstyle='steps-post', label='steps-post')
plt.legend(loc='best')
刻度、标签和图例
xlim、xticks和xticklabels之类的方法分别控制图标的范围、刻度位置、刻度标签等。
使用方法:
- 如果填写参数则按照参数进行设置
- 如果不填写则返回当前的参数值
如
plt.xlim([0, 10])
这段代码会将X轴的范围设置为0到10。也可以用ax.get_xlim
和ax.set_xlim
来达到上面的方法。
设置标题、轴标签、刻度以及刻度标签 ---- 轴的自定义
fig3 = plt.figure()
ax = fig3.add_subplot(1, 1, 1)
ax.plot(randn(1000).cumsum())
修改X轴的刻度,可以用set_xticks
和set_xticklabels
set_xticks
只能设置数据范围
ticks = ax.set_xticks([0, 250, 500, 750, 1000])
如果要完全自定义,需要用set_xticklabels
方法:
labels = ax.set_xticklabels(['one', 'two','three', 'four', 'five'], rotation=30, fontsize='small')
set_xlabel
可以为X轴设置名称,用set_title
对图表设置一个标题:
ax.set_title('My first matplotlib plot')
ax.set_label('Stages')
添加图例
图例是一种用于标识图表元素的工具
例如下面这个例子,光看图表很难区分这几条曲线,通过label参数来进行标识并且在最后通过legend
来创建图例
fig4 = plt.figure()
ax = fig4.add_subplot(1, 1, 1)
ax.plot(randn(1000).cumsum(), 'k', label='one')
ax.plot(randn(1000).cumsum(), 'k--', label='two')
ax.plot(randn(1000).cumsum(), 'k.', label='_nolegend_')
ax.legend(loc='best')
如果将label
设置为_nolegend_
,那将不显示该图例
注解以及在Subplot上绘图
注解:可以通过text、arrow、annotate等函数添加
- text : 可以在指定坐标上添加注解,并设置字体格式
ax.text(10,10,'Hello world', family='monospace', fontsize=10)
图形绘制:
和图表一样也需要在一个fingure
中绘制。要添加图形需要创建一个shp
对象,通过add_patch
添加到subplot
中
fig5 = plt.figure()
ax = fig5.add_subplot(1, 1, 1)
rect = plt.Rectangle((0.2, 0.75), 0.4, 0.15, color='k', alpha=0.3)
circ = plt.Circle((0.7, 0.2), 0.15, color='b', alpha=0.3)
pgon = plt.Polygon([[0.15, 0.15],[0.35, 0.4],[0.2,0.6]], color='g', alpha=0.5)
ax.add_patch(rect)
ax.add_patch(circ)
ax.add_patch(pgon)
将图表保存到文件
利用plt.savefig
可以保存图表到文件中:
plt.savefig('filepath.svg')
也可以设置dpi
plt.savefig('filepath.png', dpi=400, bbox_inches='tight')