matlab - Matlab 中的单位矩阵

Question

我必须在给定参数 N 的 Matlab 中创建一个函数，它返回 N×N 单位矩阵。我不能使用循环，也不能使用 eye 或 diag 等内置函数。我尝试了以下方法:

function I = identity( n )
    I = zeros( n,n );
    p = [1:n;1:n]';
    I( p ) = 1;
end

但是，当我用 I = identity(3); 调用它时，我得到以下结果:

I =

 1     0     0
 1     0     0
 1     0     0

我不明白为什么，因为我认为 Matlab 可以使用向量作为矩阵索引，而我这样做的方式是:

p =

 1     1
 2     2
 3     3

所以当我执行 I( p ) = 1 时，第一步应该是 I( 1,1 ) = 1 然后是 I( 2,2 ) = 1 等等。我没有看到什么？

Answer 1

不使用函数，只使用 matrix indexing -

A(N,N) = 0;
A((N+1)*(0:N-1)+1) = 1

因此，函数变为-

function A = identity( N )
A(N,N) = 0;
A((N+1)*(0:N-1)+1) = 1;
end