Scala中groupBy和aggregate的一次使用

1、使用Scala实现一个方法,使得返回数组中,分别等于num,大于num,小于num的元素的个数。

      例:getNum(Array(1,2,3,4,5,6),2)结果为(2,2,3)

实现方法一

val tum1=arr.filter(x => x x==num).size
val tum3=arr.filter(x => x>num).size
val tum=Tuple3(tum1,tum2,tum3)

实现方法二

val tuple: (Int, Int, Int) = arr.aggregate(0, 0 ,0)((x, y) => ((
if(ynum) {x._3+1}else{x._3})), 
(x, y) => (x._1 + y._1, x._2 + y._2,x._3+y._3 ))

实现方法三

val array: Array[Int] = arr.groupBy(x => x match {
           case x if (x < num) => "x<"+num
           case x if (x > num) => "x>2"+num
           case _ => "x="+num
         }).map(x => x._2.size).toArray

 

你可能感兴趣的:(Scala相关)