Question

How to find the sum of each row and each column of a matrix (2D array) in Scala

Answer 1

object Main extends App {
  val arr = Array(
    Array(1, 2, 3, 5),
    Array(4, 5, 6, 5),
    Array(7, 8, 9, 5)
  )

  val rows = arr.length
  val cols = arr(0).length

  for (i <- 0 until rows) {
    val sumRow = arr(i).sum
    println(s"Sum of row: $i = $sumRow")
  }

  for (i <- 0 until cols) {
    val sumCol = (0 until rows).map(j => arr(j)(i)).sum
    println(s"Sum of col: $i = $sumCol")
  }
}




/*
run:
  
Sum of row: 0 = 11
Sum of row: 1 = 20
Sum of row: 2 = 29
Sum of col: 0 = 12
Sum of col: 1 = 15
Sum of col: 2 = 18
Sum of col: 3 = 15
  
*/

 

70+ SQL courses for beginners and professionals