Scala中groupBy和aggregate的一次使用

mac2024-08-16  60

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

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

实现方法一

val tum1=arr.filter(x => x<num).size val tum2=arr.filter(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(y<num) {x._1+1}else{x._1}), (if(y==num) {x._2+1}else{x._2}), (if(y>num) {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

 

最新回复(0)