如何创建 4D 复杂曲面图？

Question

我有以下 Matlab 代码，我想将其转换为 Python 3 代码。

r = (0:1:15)';                           % create a matrix of complex inputs

theta = pi*(-2:0.05:2);

z = r*exp(1i*theta);

%w = z.^(1/2)  ;                          % calculate the complex outputs

w = sqrt(r)*exp(1i*theta/2);

figure('Name','Graphique complexe','units','normalized','outerposition',[ 0.08 0.1 0.8 0.55]);

subplot(121)

surf(real(z),imag(z),real(w),imag(w))    % visualize the complex function using surf

xlabel('Real(z)')

ylabel('Imag(z)')

zlabel('Real(u)')

cb = colorbar;

colormap jet;                            % gradient from blue to red

cb.Label.String = 'Imag(v)';

subplot(122)

surf(real(z),imag(z),imag(w),real(w))    % visualize the complex function using surf

xlabel('Real(z)')

ylabel('Imag(z)')

zlabel('Imag(v)')

cb = colorbar;

colormap jet;                            % gradient from blue to red

cb.Label.String = 'Real(u)';

可以在此处找到结果和原始讨论。此SO 页面上还有可用的讨论。但是，我未能运行和重现这些代码。接下来我可以尝试什么？