Scala集合：数组groupBy和返回每个组的数组索引

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

导读：我有一个数组,类似的东西： val a = Array("a","c","z","b","a") 我希望得到一个包含该数组的所有不同值的键的映射,以及每个这样的组的相关索引的集合,即对于给定的数组,答案是： Map( "a" - Array(0,6),"b" - Array(5),"c" - Array(1,2,4),"z" - Array(3))

我有一个数组,类似的东西：

val a = Array("a","c","z","b","a")

我希望得到一个包含该数组的所有不同值的键的映射,以及每个这样的组的相关索引的集合,即对于给定的数组,答案是：

Map(
  "a" -> Array(0,6),"b" -> Array(5),"c" -> Array(1,2,4),"z" -> Array(3)
)

令人惊讶的是,事实证明它比我预想的要复杂一些.我到目前为止最好的是：

a.zipWithIndex.groupBy {
  case(cnt,idx) => cnt
}.map {
  case(cnt,arr) => (cnt,arr.map {
    case(k,v) => v
  }
}

这既不简洁也不易理解.有更好的想法吗？

解决方法

您的代码可以重写为oneliner,但它看起来很难看.

as.zipWithIndex.groupBy(_._1).mapValues(_.map(_._2))

另一种方法是使用mutable.MultiMap

import collection.mutable.{ HashMap,MultiMap,Set }

val as = Array("a","a")
val mm = new HashMap[String,Set[Int]] with MultiMap[String,Int]

然后只需添加每个绑定

as.zipWithIndex foreach (mm.addBinding _).tupled    
//mm = Map(z -> Set(3),b -> Set(5),a -> Set(0,c -> Set(1,4))

如果你想要不可变版本,最后你可以将它转换为mm.toMap.

（编辑：李大同）

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