在JavaScript中构造阶数为n的单位矩阵

单位矩阵

单位矩阵是一个n×n方阵，对角线由1组成，其他元素全为零。

例如，顺序的单位矩阵为-

const arr = [
   [1, 0, 0],
   [0, 1, 0],
   [0, 0, 1]
];

我们需要编写一个JavaScript函数，该函数接受一个数字，例如n，并返回n * n阶的单位矩阵。

示例

以下是代码-

const num = 5;
const constructIdentity = (num = 1) => {
   const res = [];
   for(let i = 0; i < num; i++){
      if(!res[i]){
         res[i] = [];
      };
      for(let j = 0; j < num; j++){
         if(i === j){
            res[i][j] = 1;
         }else{
            res[i][j] = 0;
         };
      };
   };
   return res;
};
console.log(constructIdentity(num));

输出结果

以下是控制台上的输出-

[
   [ 1, 0, 0, 0, 0 ],
   [ 0, 1, 0, 0, 0 ],
   [ 0, 0, 1, 0, 0 ],
   [ 0, 0, 0, 1, 0 ],
   [
      0,
      0,
      0,
      0,
      1
   ]
]