在 Scala 中对映射值进行求和和分组

Question

我是 Scala 的新手，我正在寻找一种更简洁的方法来对映射值进行求和和分组。有没有比下面的代码更好的方法：

def mapSum(thisMap: Map[Char, Int], thatMap: Map[Char, Int]) = {
  thisMap.transform { (k, v) => thatMap(k) + v }
}

满足以下测试：

@Test
def mapSum() {
  val map: Map[Char, Int] = Map('C' -> 1, 'D' -> 3)
  val newMap = mapSum(map, map)
  assertEquals(2, newMap('C'))
  assertEquals(6, newMap('D'))
}

Answer 1

如果您想要简洁，使用标准库的当前版本不会比您的当前版本好多少（但请注意，您可以去掉外括号以使其成为两行代码）。

Scalaz 提供了一些工具，可以让这种事情变得更简洁一些，包括Map 的一个幺半群实例和一个拉皮条的unionWith 方法：

scala> import scalaz._, Scalaz._
import scalaz._
import Scalaz._

scala> val m = Map('C' -> 1, 'D' -> 3)
m: scala.collection.immutable.Map[Char,Int] = Map(C -> 1, D -> 3)

scala> m |+| m
res0: scala.collection.immutable.Map[Char,Int] = Map(C -> 2, D -> 6)

scala> (m unionWith m)(_ + _)
res1: Map[Char,Int] = Map(C -> 2, D -> 6)

请注意，这两种方法的行为与您的略有不同 - 如果第一个中的键不在第二个中，它们不会在运行时阻塞，并且它们不会默默地忽略第二个不在第一个中。