python画图的模块_【Python开发】使用pyplot模块绘图

使用pyplot模块绘图¶

matplotlib的pyplot模块提供了和MATLAB类似的绘图API，方便用户快速绘制二维图表。我们先看一个简单的例子：

import numpy as np

import matplotlib.pyplot as plt ❶

x = np.linspace(0, 10, 1000)

y = np.sin(x)

z = np.cos(x**2)

plt.figure(figsize=(8,4)) ❷

plt.plot(x,y,label="$sin(x)$",color="red",linewidth=2) ❸

plt.plot(x,z,"b--",label="$cos(x^2)$") ❹

plt.xlabel("Time(s)") ❺

plt.ylabel("Volt")

plt.title("PyPlot First Example")

plt.ylim(-1.2,1.2)

plt.legend()

plt.show() ❻

使用pyplot模块快速将数据绘制成曲线图

❶首先载入matplotlib的绘图模块pyplot，并且重命名为plt。

pylab模块

matplotlib还提供了一个名为pylab的模块，其中包括了许多NumPy和pyplot模块中常用的函数，方便用户快速进行计算和绘图，十分适合在IPython交互式环境中使用。本书使用下面的方式载入pylab模块：

>>>import pylab as pl

❷调用figure()创建一个Figure(图表)对象，并且它将成为当前Figure对象。也可以不创建Figure对象直接调用接下来的plot()进行绘图，这时matplotlib会自动创建一个Figure对象。figsize参数指定Figure对象的宽度和高度，其单位为英寸。此外还可以用dpi参数指定Figure对象的分辨率，即每英寸所表示的像素数，这里使用缺省值80。因此本例中所创建的Figure对象的宽度为“8*80 = 640”个像素。但是在显示出绘图窗口之后，用工具栏中的保存按钮将图表保存为图像时，所保存的图像的大小是“800*400”像素。这是因为保存图像时会使用不同的dpi设置。这个设置保存在matplotlib的配置文件中，我们可以通过如下语句查看它的值：

>>>import matplotlib

>>>matplotlib.rcParams["savefig.dpi"]

100

因为保存图像时的DPI设置为100，因此所保存的图像的宽度是“8*100 = 800”个像素。rcParams是一个字典，其中保存着从配置文件读入的所有配置，在调用各种绘图函数时，这些配置将会作为各种参数的缺省值。后面我们还会对matplotlib的配置文件进行详细介绍。

❸创建Figure对象之后，接下来调用plot()在当前的Figure对象中绘图。实际上plot()是在Axes(子图)对象上绘图，如果当前的Figure对象中没有Axes对象，将会为之创建一个几乎充满整个图表的Axes对象，并且使此Axes对象成为当前的Axes对象。plot()的前两个参数是分别表示X、Y轴数据的对象，这里使用的是NumPy数组。使用关键字参数可以指定所绘制的曲线的各种属性：

label：给曲线指定一个标签名称，此标签将在图示中显示。如果标签字符串的前后有字符’$’，则matplotlib会使用其内嵌的LaTex引擎将其显示为数学公式。

color：指定曲线的颜色，颜色可以用英文单词，或者以’#’字符开头的三个16进制数，例如’#ff0000’表示红色。或者使用值在0到1范围之内的三个元素的元组表示，例如(1.0, 0.0, 0.0)也表示红色。

linewidth：指定曲线的宽度，可以不是整数，也可以使用缩写形式的参数名lw。

使用LaTex语法绘制数学公式会极大地降低图表的描绘速度。

❹直接通过第三个参数’b–’指定曲线的颜色和线型，它通过一些易记的符号指定曲线的样式。其中’b’表示蓝色，’–’表示线型为虚线。在IPython中输入“plt.plot?”可以查看格式化字符串以及各个参数的详细说明。

❺接下来通过一系列函数设置当前Axes对象的各个属性：

xlabel、ylabel：分别设置X、Y轴的标题文字。

title：设置子图的标题。

xlim、ylim：分别设置X、Y轴的显示范围。

legend：显示图示，即图中表示每条曲线的标签(label)和样式的矩形区域。

❻最后调用plt.show()显示出绘图窗口。在通常的运行情况下，show()将会阻塞程序的运行，直到用户关闭绘图窗口。然而在带“-wthread”等参数的IPython环境下，show()不会等待窗口关闭。

还可以调用plt.savefig()将当前的Figure对象保存成图像文件，图像格式由图像文件的扩展名决定。下面的程序将当前的图表保存为“test.png”，并且通过dpi参数指定图像的分辨率为120，因此输出图像的宽度为“8*120 = 960”个像素。

>>>run matplotlib_simple_plot.py

>>>plt.savefig("test.png", dpi=120)

如果关闭了图表窗口，则无法使用savefig()保存图像。实际上不需要调用show()显示图表，可以直接用savefig()将图表保存成图像文件。使用这种方法可以很容易编写批量输出图表的程序。

savefig()的第一个参数可以是文件名，也可以是和Python的文件对象有相同调用接口的对象。例如可以将图像保存到StringIO对象中，这样就得到了一个表示图像内容的字符串。这里需要使用fmt参数指定保存的图像格式。

>>>from StringIO import StringIO

>>>buf = StringIO() # 创建一个用来保存图像内容的StringIO对象

>>>plt.savefig(buf, fmt="png") # 将图像以png格式保存进buf中

>>>buf.getvalue()[:20] # 显示图像内容的前20个字节

'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x03 '