Question

我正在尝试将一些代码从Matlab脚本移植到python，并且我在3d矩阵上运行。我正在尝试在python上实现它，但是我在python代码中遇到了一个奇怪的行为。我在这里介绍我的问题的简化版本：

（它必须是矩阵而不是数组，因为在原始问题中它必须存储符号表达式）

Matlab代码，它将创建一系列大小为 i x i 的三角形数组，其中 i = 1：n：

n = 3;
f = zeros(n+1,n+1,n+1);
for m = 1:n
    for i = 1:m+1
        for j = 1:m+1 -i +1
            f(i,j,m+1) = i*j*m;
        end
    end
end

我的python代码：

n = 3;
f = [[[0 for k in range(n+1)] for j in range(n+1)] for i in range(n+1)]
for m in range(n):
    for i in range(m+2):
        for j in range(m+2-i):
            f[i][j][m+1] = (i+1)*(j+1)*(m+1);

最后2个矩阵的Matlab输出：

f(:,:,3) =                  f(:,:,4) =

 2     4     6     0         3     6     9    12
 4     8     0     0         6    12    18     0
 6     0     0     0         9    18     0     0
 0     0     0     0         12     0     0     0

最后2个矩阵的Python输出：

[[0, 0, 6, 9], [0, 0, 0, 18], [0, 0, 0, 0], [0, 0, 0, 0]],
[[0, 0, 0, 12], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]]

我检查了两个代码中的迭代次数是否相同。我还检查了每次迭代中索引的值是否正确。例如：Matlab m，i，j =（1,1,1）等于Python m，i，j =（0,0,0）

Answer 1

好吧，切换1st和3d索引似乎解决了这个问题。正确的python代码是：

n = 3;
f = [[[0 for k in range(n+1)] for j in range(n+1)] for i in range(n+1)]
for m in range(n):
    for i in range(m+2):
        for j in range(m+2-i):
            f[m+1][i][j] = (i+1)*(j+1)*(m+1);

，输出确认它：

 [[2, 4, 6, 0], [4, 8, 0, 0], [6, 0, 0, 0], [0, 0, 0, 0]],
 [[3, 6, 9, 12], [6, 12, 18, 0], [9, 18, 0, 0], [12, 0, 0, 0]]]

将3D矩阵从Matlab移植到python

1 个答案: