central tendency measurements calculator

CentralTendencyMeasurements.scala

import scala.util.Sorting
import scala.collection.mutable.Map

@main
def main(): Unit = {
  val list = List(26,0,5,8,20,22,0,13,6)
  println(list.toCentralTendencyMeasures())
}


case class CentralTendencyMeasures(mode: Int, median: Double, mean: Double)

extension(list: Iterable[Int])
  def toCentralTendencyMeasures(): CentralTendencyMeasures = {
    val sortedList = Sorting.stableSort(list.toSeq).toList
    var counts = Map[Int, Int]()
    val map = sortedList.groupBy(identity).mapValues(_.size)
    val mode = map.maxBy(pair => pair._2)._1
    var median = 0.0
    if sortedList.size % 2 == 0 then
      median = (sortedList(sortedList.size / 2 - 1) + sortedList(sortedList.size / 2)) / 2.0
    else
      median = sortedList(sortedList.size / 2)
    val mean = sortedList.sum.toDouble / sortedList.size
    CentralTendencyMeasures(mode, median, mean)
  }