在Scala中，是否有相当于Haskell的“fromListWith”用于Map？

Question

在Haskell中，有一个名为fromListWith的函数，它可以从函数生成Map（用于合并具有相同键的值）和列表：

fromListWith :: Ord k => (a -> a -> a) -> [(k, a)] -> Map k a

以下表达式将被评估为true ：

fromListWith (++) [(5,"a"), (5,"b"), (3,"b"), (3,"a"), (5,"a")] == fromList [(3, "ab"), (5, "aba")]

在Scala中，在List对象上有一个名为toMap的类似函数，它也可以将列表转换为Map，但是它不能具有处理重复键的函数参数。

有没有人有这个想法？

Answer 1

除了使用scalaz之外，您还可以自己定义一个：

implicit class ListToMapWith[K, V](list: List[(K, V)]) {
  def toMapWith(op: (V, V) => V) = 
    list groupBy (_._1) mapValues (_ map (_._2) reduce op)
}

这是一个用法示例：

scala> val testList = List((5,"a"), (5,"b"), (3,"b"), (3,"a"), (5,"a"))
scala> testList toMapWith (_ + _)
res1: scala.collection.immutable.Map[Int,String] = Map(5 -> aba, 3 -> ba)

Answer 2

该STDLIB不具备这样的功能，但是，没有一个端口Data.Map可用scalaz这确实有提供此功能。