Python数值计算:一 使用Pylab绘图(3)

1.4.1图像的画法

杨氏双缝干涉图样（Double-slit experiment）

在杨氏双缝干涉实验中，光强的公式为：

\[I = 4I_0Cos^2\frac{\Delta\phi}2 \]

其中，

\[I_0 \]

是一条缝的光强，

\[\Delta\phi \]

是两束光相遇时的相伴差。画出光的干涉图样。

n = 3
dphi = arange(-1,1,0.01) * n *2 *pi
i = 4 * cos(dphi/2) **2
 #
subplot(211)
plot(dphi,i)
 #
subplot(212)
imshow(vstack((i,i)),cmap=mpl.cm.gray, extent=(-20,20,0,8)) #1

(Tips)1：直接使用一维（1D）数组无法在matplotlib中绘图，需要将两个一维数组合成为一个二维（2D）数组。vstack是numpy提供的矩阵组合函数，可以将两个一维数组按行组合；column_stack则是按列组合，两个函数的参数是一个tuple，所以需要用逗号+括号标明。
2:cmap是colormap的缩写，可以自己指定需要的配色。
3:extent指定坐标轴的大小。