Scala中groupBy和aggregate的一次使用

最新推荐文章于 2022-09-24 21:30:24 发布

东北溜达滑

最新推荐文章于 2022-09-24 21:30:24 发布

阅读量857

点赞数

分类专栏： Scala相关文章标签： Scala

本文链接：https://blog.csdn.net/qq_39188747/article/details/102844922

版权

Scala相关专栏收录该内容

5 篇文章 0 订阅

订阅专栏

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