在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
]
]