scala - 在Scala中，我该如何做相当于SQL SUM和GROUP BY？

Question

例如，假设我有

val list: List[(String, Double)]

有值(value)观

"04-03-1985", 1.5
"05-03-1985", 2.4
"05-03-1985", 1.3

我如何产生新 list

"04-03-1985", 1.5
"05-03-1985", 3.7

Answer 1

这是一线。除非确实将这些高阶函数的类型内部化，否则它不是特别可读。

val s = Seq(("04-03-1985" -> 1.5),
            ("05-03-1985" -> 2.4),
            ("05-03-1985" -> 1.3))

s.groupBy(_._1).mapValues(_.map(_._2).sum)
// returns: Map(04-03-1985 -> 1.5, 05-03-1985 -> 3.7)

另一种方法是使用折叠一对一地添加键值对，

s.foldLeft(Map[String, Double]()) { case (m, (k, v)) =>
  m + (k -> (v + m.getOrElse(k, 0d)))
}

我认为，理解力最强的途径是

var m = Map[String, Double]()
for ((k, v) <- s) {
  m += k -> (v + m.getOrElse(k, 0d))
}

也许使用Scalaz的Map的monoid类型类可以做得更好。

请注意，您可以使用 Map[K, V]和 Seq[(K, V)]方法在 toSeq和 toMap之间转换。

更新。经过更多考虑之后，我认为自然抽象将是类型的“多图”转换，

def seqToMultimap[A, B](s: Seq[(A, B)]): Map[A, Seq[B]]

在个人图书馆中使用适当的隐式扩展，然后可以编写:

s.toMultimap.mapValues(_.sum)

我认为这是最清晰的!