将列表转换为元素映射 - > scala中的list(元素)

时间:2011-12-30 03:17:53

标签: scala collections higher-order-functions

我有一份文件清单,其中一个文件的所有者是一个用户。

将此列表转换为用户地图到他们拥有的文档列表的最优雅方法是什么?

例如,我有:

"doc1" owned by user "John"
"doc2" owned by user "Frank"
"doc3" owned by user "John"

我应该得到一张地图:

"John" -> List("doc1", "doc3"), "Frank" -> List("doc2")

我可以想到一种方法,即从文档中获取所有唯一用户,并且每个用户都将文档列表过滤为他们拥有的文档列表,但我想知道是否有使用固定文件的方法通过列表的次数,以防止列表很大时出现任何性能问题。

1 个答案:

答案 0 :(得分:13)

使用groupBy:

scala> case class Doc(id: String, owner: String)
defined class Doc

scala> List(Doc("doc1", "John"), Doc("doc2", "Frank"), Doc("doc3", "John"))
res0: List[Doc] = List(Doc(doc1,John), Doc(doc2,Frank), Doc(doc3,John))

scala> res0.groupBy(_.owner)
res1: scala.collection.immutable.Map[String,List[Doc]] = Map(
  Frank -> List(Doc(doc2,Frank)), John -> List(Doc(doc1,John), Doc(doc3,John)))