RosettaCodeData/Task/Identity-matrix/Scala/identity-matrix.scala
2023-07-01 13:44:08 -04:00

4 lines
193 B
Scala

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))