我说的二维是指做成那种切面，画出来的二维图，能不能据此修改一下

非常抱歉理解错误了你的意思。如果你需要绘制二维波前场的切面，可以使用 Matlab 中的 contour 函数。以下是一个修改后的代码：

% 设置波长和波数
lambda = 0.1; % 波长
k = 2*pi/lambda; % 波数

% 设置网格大小和步长
Lx = 10; % x 轴范围
Ly = 10; % y 轴范围
dx = 0.1; % x 轴步长
dy = 0.1; % y 轴步长

% 创建网格
x = 0:dx:Lx;
y = 0:dy:Ly;
[X,Y] = meshgrid(x,y);

% 创建波函数
A = 1; % 波的振幅
phi = pi/2; % 波的相位
psi = k*X*cos(phi) + k*Y*sin(phi); % 波函数
U = A*cos(psi);

% 创建反射和折射波函数
theta1 = pi/4; % 入射角
theta2 = asin(sin(theta1)/1.5); % 折射角
theta3 = pi - theta1; % 反射角
psi_r = k*X*cos(theta3) - k*Y*sin(theta3); % 反射波函数
psi_t = k*X*cos(theta2) + k*Y*sin(theta2); % 折射波函数
U_r = A*cos(psi_r); % 反射波
U_t = A*cos(psi_t); % 折射波

% 创建底部不平坦的表面
Z = -0.5*(X.^2 + Y.^2); % 圆形表面
Z(Z<-2) = -2; % 平坦表面
Z = Z + 0.5; % 离开表面

% 计算反射和折射后的波前场
U_r(Z>-0.5) = 0; % 反射波在表面以下不显示
U_t(Z<-0.5) = 0; % 折射波在表面以上不显示
U_total = U + U_r + U_t;

% 绘制切面图像
figure;
contour(X,Y,U_total,'ShowText','on');
xlabel('x');
ylabel('y');
title('二维波前场切面图（带有反射和折射）');

在这个代码中，我们使用了 contour 函数来绘制切面图像。该函数会根据二维网格和波前场的数值自动绘制出等值线，并在等值线上标注数值。可以通过修改 contour 函数的参数来控制等值线的数量和颜色，以及是否显示数值标注等。