javascript - 使用高阶函数的 N*N 大小的单位矩阵

Question

我正在尝试仅使用Array.prototype中的方法创建一个n * n单位矩阵(欢迎您调用其他方法)。我可以使用 fill() 创建一个 n*n 矩阵，但在不使用 if-else 的情况下将对 Angular 线元素转换为 1 时遇到问题。

function matrix(n) {
  var nSizeArray = Array(n).fill().map(function(elem) {
    return Array(elem).fill(0).indexOf([elem][elem]=1)
  })
  return nSizeArray
}
matrix(5)

以上只是我想要做的一些演示。我无法理解如何使用可用的数组方法继续填充数组并访问数组中的 [i][i] 元素。

Answer 1

您可以使用以下函数创建n * n单位矩阵:

const eye = n => [...Array(n)].map((e, i, a) => a.map(e => +!i--));

console.log(eye(3));

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