RosettaCodeData/Task/Identity-matrix/Scala/identity-matrix.scala

5 lines
193 B
Scala
Raw Permalink Normal View History

2023-07-01 11:58:00 -04:00
def identityMatrix(n:Int)=Array.tabulate(n,n)((x,y) => if(x==y) 1 else 0)
def printMatrix[T](m:Array[Array[T]])=m map (_.mkString("[", ", ", "]")) mkString "\n"
printMatrix(identityMatrix(5))