Scala:按键合并地图

Question

说我有两张地图:

val a = Map(1 -> "one", 2 -> "two", 3 -> "three")
val b = Map(1 -> "un", 2 -> "deux", 3 -> "trois")

我想通过键合并这些映射,应用一些函数来收集值(在这种特殊情况下,我想将它们收集到一个seq中,给出:

val c = Map(1 -> Seq("one", "un"), 2 -> Seq("two", "deux"), 3 -> Seq("three", "trois"))

感觉应该有一个很好的,惯用的方式来做这个 - 任何建议？如果解决方案涉及scalaz,我很高兴.

Answer 1

scala.collection.immutable.IntMap有一种intersectionWith方法可以精确地做你想要的(我相信):

import scala.collection.immutable.IntMap

val a = IntMap(1 -> "one", 2 -> "two", 3 -> "three", 4 -> "four")
val b = IntMap(1 -> "un", 2 -> "deux", 3 -> "trois")

val merged = a.intersectionWith(b, (_, av, bv: String) => Seq(av, bv))

这给了你IntMap(1 -> List(one, un), 2 -> List(two, deux), 3 -> List(three, trois)).请注意,它正确地忽略了仅出现的键a.

作为一个方面说明:我经常发现自己想要的unionWith,intersectionWith从等功能Haskell的Data.Map斯卡拉.我认为没有任何原则性的理由,它们只应该IntMap在基础collection.Map特征上,而不是在基本特征中.

Answer 2

val a = Map(1 -> "one", 2 -> "two", 3 -> "three")
val b = Map(1 -> "un", 2 -> "deux", 3 -> "trois")

val c = a.toList ++ b.toList
val d = c.groupBy(_._1).map{case(k, v) => k -> v.map(_._2).toSeq}
//res0: scala.collection.immutable.Map[Int,Seq[java.lang.String]] =
        //Map((2,List(two, deux)), (1,List(one, un), (3,List(three, trois)))

Answer 3

Scalaz增加了一个方法|+|的任何类型的A针对其Semigroup[A]可用.

如果您映射地图以使每个值都是单个元素序列,那么您可以非常简单地使用它:

scala> a.mapValues(Seq(_)) |+| b.mapValues(Seq(_))
res3: scala.collection.immutable.Map[Int,Seq[java.lang.String]] = Map(1 -> List(one, un), 2 -> List(two, deux), 3 -> List(three, trois))