scala - map 上的 FlatmapValues

Question

给定一个 Seq 元组，例如:

Seq(
  ("a",Set(1,2)),
  ("a",Set(2,3)),
  ("b",Set(4,6)),
  ("b",Set(5,6))
)

我想groupBy然后flatMap值来获得类似的东西:

Map(
  b -> Set(4, 6, 5), 
  a -> Set(1, 2, 3)
)

我的第一个实现是:

Seq(
  ("a" -> Set(1,2)),
  ("a" -> Set(2,3)),
  ("b" -> Set(4,6)),
  ("b" -> Set(5,6))
) groupBy (_._1) mapValues (_ map (_._2)) mapValues (_.flatten.toSet)

我想知道是否有更有效且可能更简单的方法来实现该结果。

Answer 1

您的方向是正确的，但您可以通过使用单个 mapValues 并结合 map 和 flatten 来简化一点:

val r = Seq(
  ("a" -> Set(1,2)),
  ("a" -> Set(2,3)),
  ("b" -> Set(4,6)),
  ("b" -> Set(5,6))
).groupBy(_._1).mapValues(_.flatMap(_._2).toSet)

我实际上发现它比 foldLeft 版本更具可读性(但请注意 mapValues 返回 a non-strict collection ，这可能是也可能不是您想要的)。