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

Question

我有以下 Matlab 代码，我想将其转换为 Python 3 代码。r = (0:1:15)';                           % create a matrix of complex inputstheta = pi*(-2:0.05:2);z = r*exp(1i*theta);%w = z.^(1/2)  ;                          % calculate the complex outputsw = 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 surfxlabel('Real(z)')ylabel('Imag(z)')zlabel('Real(u)')cb = colorbar;colormap jet;                            % gradient from blue to redcb.Label.String = 'Imag(v)';subplot(122)surf(real(z),imag(z),imag(w),real(w))    % visualize the complex function using surfxlabel('Real(z)')ylabel('Imag(z)')zlabel('Imag(v)')cb = colorbar;colormap jet;                            % gradient from blue to redcb.Label.String = 'Real(u)';可以在此处找到结果和原始讨论。此SO 页面上还有可用的讨论。但是，我未能运行和重现这些代码。接下来我可以尝试什么？