arrays - Scala 二维数组按指定列排序

Question

给定任何二维数组，例如，

val in = Array( Array(59, 45, 32), 
                Array(20, 88, 5), 
                Array(49, 72, 89))

想按第三列排序，以便

val out = Array( Array(20, 88, 5),
                 Array(59, 45, 32),                       
                 Array(49, 72, 89))

非常感谢。

Answer 1

这应该可以工作，但如果您的数组大小小于 3，则会失败:

scala> in.sortBy(_(2))

输出:

res0: Array[Array[Int]] = Array(Array(20, 88, 5), Array(59, 45, 32), Array(49, 72, 89))

您也可以像这样设置故障保护:

in.sortBy {
  case arr @ Array(_) if arr.size >= 3 => arr(2)
  case arr => Int.MaxValue // or any other value. Using MaxValue will make the invalid arrays be last ones
}