Как создать изменчивую матрицу идентичности в Scala?

Question

Я должен создать идентификационную матрицу в Scala и без использования zipWithIndex (я успешно пытался использовать zipWithIndex, но это вызвало некоторые проблемы из-за полученного IndexedSeq). Тип этого должен быть: mutable.Seq[mutable.Seq[Double]].

Как я мог это сделать?

Answer 1

Попробуйте это:

import scala.collection.mutable.Seq
def identity(dim: Int): Seq[Seq[Double]] = {
  Seq.tabulate(dim)(r => Seq.tabulate(dim)(c => if(r == c) 1.0 else 0.0))
}