python - 如何用python重写矩阵3D matlab脚本

Question

我是Python新手。我有如下所示的 matlab 脚本。我想将 matlab 脚本中的 3D 矩阵重写为 python 3.x 语言。我该如何修复它？

nl=length(res);
ndat=length(per);
phi=atan(1)*4;
amu=phi*4e-7;

for i=1:ndat
  for j=1:nl
     z=sqrt(phi*amu*res(j)/per(i));
     zz(j)=complex(z,z);
     exp0=exp((-2)*zz(j)/res(j)*thi(j));
     exp1=complex(1,0)+exp0;
     exp2=complex(1,0)-exp0;

     %matrix 3D
     ldi(1,1,j)=exp1;
     ldi(1,2,j)=zz(j)*exp2
     ldi(2,1,j)=exp2/zz(j);
     ldi(2,2,j)=exp1;`
  end
end

Answer 1

您将找到代码的独立实现(如下)，但有一些关键区别:

1. Python 索引从 0 开始，而不是从 1 开始

2. Python 索引使用方括号而不是圆括号

3. 数学函数必须从库中导入(此处为 math 和 cmath)

祝你好运!

import math
import cmath

# Data
res = [1, 4, 1, 2, 3]
per = [5, 5, 1, 1, 0.5, 0.6]
thi = [1, 2, 3, 4, 5, 6]
nl = len(res)
ldi = [[[0 for x in range(nl)],[0 for x in range(nl)]], [[0 for x in range(nl)],[0 for x in range(nl)]]]
zz = [0]*nl


nl = len(res)
ndat = len(per)
phi = math.atan(1)*4
amu = phi*4e-7

for i in range(ndat):
  for j in range(nl):
    z = math.sqrt(phi*amu*res[j]/per[i])
    zz[j] = complex(z,z)
    exp0=cmath.exp((-2)*zz[j]/res[j]*thi[j]);
    exp1=complex(1,0)+exp0;
    exp2=complex(1,0)-exp0;

    #- matrix 3D
    ldi[0][0][j]=exp1;
    ldi[0][1][j]=zz[j]*exp2
    ldi[1][0][j]=exp2/zz[j]
    ldi[1][1][j]=exp1