扩展分组

Question

我有一个对象集合，每个对象里面都有一个列表，让我们说：

case class Article(text: String, labels: List[String])

我需要构建一个映射，使每个键对应一个标签，其关联值是具有该标签的文章列表。

我希望下面的例子能让我的目标更加明确。我想转换以下列表

List(
  Article("article1", List("label1", "label2")),
  Article("article2", List("label2", "label3"))
)

变成Map[String,List[Article]]:

Map(
  "label1" -> List(Article("article1", ...)),
  "label2" -> List(Article("article1", ...), Article("article2", ...)),
  "label3" -> List(Article("article2", ...))
)

是否有一种优雅的方式来执行此转换？（我的意思是使用集合方法而不需要直接使用可变集合）

Answer 1

这个怎么样？

val ls = List(
  Article("article1", List("label1", "label2")),
  Article("article2", List("label2", "label3"))
)

val m = ls
    .flatMap{case a @ Article(t,ls) => ls.map(_ -> a)}
    .groupBy(_._1)
    .mapValues(_.map(_._2))

m.foreach(println)
// (label2,List(Article(article1,List(label1, label2)), Article(article2,List(label2, label3))))
// (label1,List(Article(article1,List(label1, label2))))
// (label3,List(Article(article2,List(label2, label3))))

扩展分组

Extended groupBy

scala

scala-collections