的位置给两条函数曲线加上一个注释。首先,我们在对应的函数图像位置上画一个点;然后,向横轴引一条垂线,以虚线标记;最后,写上标签。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| ...
t = 2*np.pi/3
plot([t,t],[0,np.cos(t)], color ='blue', linewidth=2.5, linestyle="--")
scatter([t,],[np.cos(t),], 50, color ='blue')
annotate(r'$\sin(\frac{2\pi}{3})=\frac{\sqrt{3}}{2}$',
xy=(t, np.sin(t)), xycoords='data',
xytext=(+10, +30), textcoords='offset points', fontsize=16,
arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))
plot([t,t],[0,np.sin(t)], color ='red', linewidth=2.5, linestyle="--")
scatter([t,],[np.sin(t),], 50, color ='red')
annotate(r'$\cos(\frac{2\pi}{3})=-\frac{1}{2}$',
xy=(t, np.cos(t)), xycoords='data',
xytext=(-90, -50), textcoords='offset points', fontsize=16,
arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=.2"))
...
|
精益求精
坐标轴上的记号标签被曲线挡住了,作为强迫症患者(雾)这是不能忍的。我们可以把它们放大,然后添加一个白色的半透明底色。这样可以保证标签和曲线同时可见。
1
2
3
4
5
| ...
for label in ax.get_xticklabels() + ax.get_yticklabels():
label.set_fontsize(16)
label.set_bbox(dict(facecolor='white', edgecolor='None', alpha=0.65 ))
...
|
图像、子图、坐标轴和记号
到目前为止,我们都用隐式的方法来绘制图像和坐标轴。快速绘图中,这是很方便的。我们也可以显式地控制图像、子图、坐标轴。Matplotlib 中的「图像」指的是用户界面看到的整个窗口内容。在图像里面有所谓「子图」。子图的位置是由坐标网格确定的,而「坐标轴」却不受此限制,可以放在图像的任意位置。我们已经隐式地使用过图像和子图:当我们调用plot
函数的时候,matplotlib 调用 gca()
函数以及 gcf()
函数来获取当前的坐标轴和图像;如果无法获取图像,则会调用figure()
函数来创建一个——严格地说,是用 subplot(1,1,1)
创建一个只有一个子图的图像。
图像
所谓「图像」就是 GUI 里以「Figure #」为标题的那些窗口。图像编号从 1 开始,与 MATLAB 的风格一致,而于 Python 从 0 开始编号的风格不同。以下参数是图像的属性:
参数 | 默认值 | 描述 |
---|
num | 1 | 图像的数量 |
figsize | figure.figsize | 图像的长和宽(英寸) |
dpi | figure.dpi | 分辨率(点/英寸) |
facecolor | figure.facecolor | 绘图区域的背景颜色 |
edgecolor | figure.edgecolor | 绘图区域边缘的颜色 |
frameon | True | 是否绘制图像边缘 |
这些默认值可以在源文件中指明。不过除了图像数量这个参数,其余的参数都很少修改。
你在图形界面中可以按下右上角的 X 来关闭窗口(OS X 系统是左上角)。Matplotlib 也提供了名为 close
的函数来关闭这个窗口。close
函数的具体行为取决于你提供的参数:
- 不传递参数:关闭当前窗口;
- 传递窗口编号或窗口实例(instance)作为参数:关闭指定的窗口;
all
:关闭所有窗口。
和其他对象一样,你可以使用 setp
或者是 set_something
这样的方法来设置图像的属性。
子图
你可以用子图来将图样(plot)放在均匀的坐标网格中。用 subplot
函数的时候,你需要指明网格的行列数量,以及你希望将图样放在哪一个网格区域中。此外,gridspec
的功能更强大,你也可以选择它来实现这个功能。
坐标轴
坐标轴和子图功能类似,不过它可以放在图像的任意位置。因此,如果你希望在一副图中绘制一个小图,就可以用这个功能。
记号
良好的记号是图像的重要组成部分。Matplotlib 里的记号系统里的各个细节都是可以由用户个性化配置的。你可以用 Tick Locators 来指定在那些位置放置记号,用 Tick Formatters 来调整记号的样式。主要和次要的记号可以以不同的方式呈现。默认情况下,每一个次要的记号都是隐藏的,也就是说,默认情况下的次要记号列表是空的——NullLocator
。
Tick Locators
下面有为不同需求设计的一些 Locators。
类型 | 说明 |
---|
NullLocator | No ticks. |
IndexLocator | Place a tick on every multiple of some base number of points plotted. |
FixedLocator | Tick locations are fixed. |
LinearLocator | Determine the tick locations. |
MultipleLocator | Set a tick on every integer that is multiple of some base. |
AutoLocator | Select no more than n intervals at nice locations. |
LogLocator | Determine the tick locations for log axes. |
这些 Locators 都是 matplotlib.ticker.Locator
的子类,你可以据此定义自己的 Locator。以日期为 ticks 特别复杂,因此 Matplotlib 提供了matplotlib.dates
来实现这一功能。
其他类型的图
接下来的内容是练习。请运用你学到的知识,从提供的代码开始,实现配图所示的效果。具体的答案可以点击配图下载。
普通图
1
2
3
4
5
6
7
8
9
| from pylab import *
n = 256
X = np.linspace(-np.pi,np.pi,n,endpoint=True)
Y = np.sin(2*X)
plot (X, Y+1, color='blue', alpha=1.00)
plot (X, Y-1, color='blue', alpha=1.00)
show()
|
散点图
1
2
3
4
5
6
7
8
| from pylab import *
n = 1024
X = np.random.normal(0,1,n)
Y = np.random.normal(0,1,n)
scatter(X,Y)
show()
|
条形图
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| from pylab import *
n = 12
X = np.arange(n)
Y1 = (1-X/float(n)) * np.random.uniform(0.5,1.0,n)
Y2 = (1-X/float(n)) * np.random.uniform(0.5,1.0,n)
bar(X, +Y1, facecolor='#9999ff', edgecolor='white')
bar(X, -Y2, facecolor='#ff9999', edgecolor='white')
for x,y in zip(X,Y1):
text(x+0.4, y+0.05, '%.2f' % y, ha='center', va= 'bottom')
ylim(-1.25,+1.25)
show()
|
等高线图
1
2
3
4
5
6
7
8
9
10
11
12
| from pylab import *
def f(x,y): return (1-x/2+x**5+y**3)*np.exp(-x**2-y**2)
n = 256
x = np.linspace(-3,3,n)
y = np.linspace(-3,3,n)
X,Y = np.meshgrid(x,y)
contourf(X, Y, f(X,Y), 8, alpha=.75, cmap='jet')
C = contour(X, Y, f(X,Y), 8, colors='black', linewidth=.5)
show()
|
灰度图(Imshow)
1
2
3
4
5
6
7
8
9
| from pylab import *
def f(x,y): return (1-x/2+x**5+y**3)*np.exp(-x**2-y**2)
n = 10
x = np.linspace(-3,3,4*n)
y = np.linspace(-3,3,3*n)
X,Y = np.meshgrid(x,y)
imshow(f(X,Y)), show()
|
饼状图
1
2
3
4
5
| from pylab import *
n = 20
Z = np.random.uniform(0,1,n)
pie(Z), show()
|
量场图(Quiver Plots)
1
2
3
4
5
| from pylab import *
n = 8
X,Y = np.mgrid[0:n,0:n]
quiver(X,Y), show()
|
网格
1
2
3
4
5
6
7
8
9
| from pylab import *
axes = gca()
axes.set_xlim(0,4)
axes.set_ylim(0,3)
axes.set_xticklabels([])
axes.set_yticklabels([])
show()
|
多重网格
1
2
3
4
5
6
7
| from pylab import *
subplot(2,2,1)
subplot(2,2,3)
subplot(2,2,4)
show()
|
极轴图
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| from pylab import *
axes([0,0,1,1])
N = 20
theta = np.arange(0.0, 2*np.pi, 2*np.pi/N)
radii = 10*np.random.rand(N)
width = np.pi/4*np.random.rand(N)
bars = bar(theta, radii, width=width, bottom=0.0)
for r,bar in zip(radii, bars):
bar.set_facecolor( cm.jet(r/10.))
bar.set_alpha(0.5)
show()
|
3D 图
1
2
3
4
5
6
7
8
9
10
11
12
13
14
| from pylab import *
from mpl_toolkits.mplot3d import Axes3D
fig = figure()
ax = Axes3D(fig)
X = np.arange(-4, 4, 0.25)
Y = np.arange(-4, 4, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='hot')
show()
|
手稿
结束篇
Matplotlib 能有今天这样强大的功能和广泛的使用得益于大量的文档和社区开发者。这里提供一些有益的链接。
入门教程
Matplotlib 文档
随代码分发的文档
Matplotlib 的代码是自文档(Self-documented)的。你可以在 Python 中快速查询某个命令的用法。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
| >>> from pylab import *
>>> help(plot)
Help on function plot in module matplotlib.pyplot:
plot(*args, **kwargs)
Plot lines and/or markers to the
:class:`~matplotlib.axes.Axes`. *args* is a variable length
argument, allowing for multiple *x*, *y* pairs with an
optional format string. For example, each of the following is
legal::
plot(x, y)
plot(x, y, 'bo')
plot(y)
plot(y, 'r+')
If *x* and/or *y* is 2-dimensional, then the corresponding columns
will be plotted.
...
|
画廊
Matplotlib 画廊 也非常有用。其中的例子都有配图和对应的代码,当你不知道某一个效果如何实现的时候,你可以在这里找找。
哦,这里还有一个小一点的画廊。