scala – 如何找到数组中最常见的元素？

发布时间：2020-12-16 09:52:09 所属栏目：安全来源：网络整理

导读：如何编写一个取整数数组的函数,并返回一个整数数组.返回值应包含输入数组中最常见的整数. List(5,4,3,2,5,1,6,4) = List(5,4)List(1,7) = List(1)List(1,7) = List(1,7) 我已经尝试过以下方法： def mostFreq(info: List[List[String]]): (String,Int) = inf

如何编写一个取整数数组的函数,并返回一个整数数组.返回值应包含输入数组中最常见的整数.

List(5,4,3,2,5,1,6,4)  => List(5,4)
List(1,7)              => List(1)
List(1,7)                 => List(1,7)

我已经尝试过以下方法：

def mostFreq(info: List[List[String]]): (String,Int) = 
  info.flatten.groupBy(identity).mapValues(_.size).maxBy(_._2)

但它没有处理关系.

解决方法

您可以通过groupBy / mapValues / maxBy捕获类似于您想出的最大频率,然后将collect应用于频率Map以获取具有最大频率的元素列表,如下所示：

val list = List(5,4)

val freqMap = list.groupBy(identity).mapValues(_.size)

val maxFreq = freqMap.maxBy(_._2)._2

freqMap.collect{ case (elem,freq) if freq == maxFreq => elem }
// res1: scala.collection.immutable.Iterable[Int] = List(5,4)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!