日志

Python科学计算技巧积累五——2D colormap 绘制

已有 3910 次阅读| 2015-6-11 15:59 |个人分类:编程语言

绘制2D colormap 更能生动形象的表明两个变量对函数的影响。在文献中插入2D 图像更显专业，也有吸引力。

使用python绘制2D图像的代码如下：

XX, YY = np.meshgrid(x_angle, y_period) # 首先生成meshgrid

fig = plt.figure()

plt.pcolor(YY, XX, absp) # 绘制 x自变量，y自变量，和函数absp

plt.colorbar()

plt.axis([300, 900, 0, 85.5]) # 设置图像范围

plt.xlabel('Grating Period $ \Lambda $ (nm)', fontsize = 15)

plt.ylabel('Incident Angle ($ \circ $)', fontsize = 15)

plt.rc('xtick', labelsize = 15)

plt.rc('ytick', labelsize = 15)

plt.show()

绘制后的图像如下：