Matlab:使用矩阵运算而不是 for 循环

Question

是否可以在 Matlab 中仅使用矩阵运算来创建 NxN 矩阵 Mat，就像下面的两个 foor 循环一样？

Mat = zeros(N);
for row = 1:N
    for col = 1:N
        if (row == 1 && (1 <= col && col <= N))
            Mat(row,col) = N;
        end
        if ((2 <= row && row <= N) && (1 <= col && col <= N))
            Mat(row,col) = (2*row+1)*col;
        end
    end
end

我想像这样索引相应的行和列:

Mat(1,1:N) = N;

row = 2:N;
col = 1:N;
Mat(row,col) = (2.*row+1).*col;

第一行有效。但是第二个操作明显导致row和col的维度有问题。

如何使用行和列的每个值？或者是否有更简单的方法从两个 foor 循环中获得相同的结果？

Answer 1

你也可以使用 ndgrid ;

[II,JJ] = ndgrid(1:N);
Mat = JJ+2*JJ.*II;
Mat(1,:) = N;