Scala集合上的高效分组聚合

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

导读：我经常需要做类似的事情 coll.groupBy(f(_)).mapValues(_.foldLeft(x)(g(_,_))) 实现相同效果的最佳方法是什么,但避免使用groupBy显式构建中间集合？解决方法您可以在包含中间结果的地图上折叠初始集合： def groupFold[A,B,X](as: Iterable[A],f: A = B,i

我经常需要做类似的事情

coll.groupBy(f(_)).mapValues(_.foldLeft(x)(g(_,_)))

实现相同效果的最佳方法是什么,但避免使用groupBy显式构建中间集合？

解决方法

您可以在包含中间结果的地图上折叠初始集合：

def groupFold[A,B,X](as: Iterable[A],f: A => B,init: X,g: (X,A) => X): Map[B,X] = 
  as.foldLeft(Map[B,X]().withDefaultValue(init)){
    case (m,a) => {
      val key = f(a)
      m.updated(key,g(m(key),a))
    }
  }

你说收藏和我写了Iterable,但你必须考虑在你的问题中顺序是否重要.

如果你想要高效的代码,你可能会像在Rex的回答中那样使用可变映射.

（编辑：李大同）

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