RosettaCodeData/Task/Identity-matrix/Scala/identity-matrix.scala
Ingy döt Net db842d013d A-M baby
2013-04-10 21:29:02 -07: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))