У меня проблема с преобразованием этого кода Matlab в python. w2_A, w2_B и w2 плюс Tk [1] и Tk [2] дают те же результаты, что и в коде Matlab, но когда код достигает Tk [m] результаты немного отличаются, и это вызывает проблемы в моем коде после. Я пробовал много разных способов конвертировать код, но получаю такие же разные результаты. В чем может быть проблема?
К вашему сведению, nmax и N равны 128
Код Matlab
nmax= 128
N= 128
x = 0:N - 1;
w = 2*x-N+1;
w1= sqrt((N*N-1)/3);
Tk( 1, : ) = ones( 1, length( x ) )/sqrt(N);
Tk( 2, : ) = (w ./ w1).*Tk( 1, : );
for m = 3:nmax
ni=m-1;
w2_A = N^2-ni^2;
w2_B = (2*ni+1)*(2*ni-1);
w2 = ni*sqrt(w2_A/w2_B);
Tk(m,:) = w./w2.*Tk(m-1,:) - w1/w2*Tk(m-2,:);
end
Код Python
nmax= 128
N= 128
x = np.arange(0, N)
w = 2*x-N+1
w1 = np.sqrt((N * N - 1) / 3)
Tk = np.zeros((nmax, len(x)))
Tk[1] = np.ones((1, len(x))) / np.sqrt(N)
Tk[2] = np.multiply(np.divide(w, w1), Tk[1])
for m in range(3, nmax):
ni = m-1
w2_A = N**2 - ni**2
w2_B = (2 * ni + 1) * (2 * ni - 1)
w2 = ni * np.sqrt(w2_A / w2_B)
Tk[m] = np.multiply(np.divide(w, w2), Tk[m - 1]) - w1 / w2 * Tk[m - 2]
Спасибо за вашу помощь, я исправил некоторые ошибки. Я почти не могу понять результаты в Tk [m]. Поскольку w2_A и w2_B абсолютно одинаковы (я это тестировал), как, например, np.divide (w, w2) может давать разные результаты, чем w./w2 в Matlab?
Пожалуйста, определите все переменные в коде (например,
N
не определен) и опубликуйте действительный код (например, в цикле Matlab отсутствуетend
), чтобы фрагменты кода можно было скопировать и вставить в командное окно. Кроме того,range(3,nmax)
Python не эквивалентен3:nmax
Matlab.