如何在不先映射它们的情况下对集合元素的字段求和(如foldLeft / reduceLeft)?

时间:2011-02-21 09:45:23

标签: scala collections sum mapreduce higher-order-functions

考虑这个课程:

 case class Person(val firstName: String, val lastName: String, age: Int)
 val persons = Person("Jane", "Doe", 42) :: Person("John", "Doe", 45) :: 
               Person("Joe", "Doe", 43) :: Person("Doug", "Don", 65) :: 
               Person("Darius", "Don", 24) :: Person("Dora", "Don", 20) :: 
               Person("Dane", "Dons", 29) :: Nil

为了得到所有人的年龄总和,我可以编写如下代码:

persons.foldLeft(0)(_ + _.age)

但是如果我想使用sum,我需要先映射值,代码如下:

persons.map(_.age).sum

如何在不使用创建某个中间集合的情况下使用sum方法

(我知道这样的“优化”很可能在没有在紧密循环中运行时没有任何真正的性能差异,而且我也知道懒惰的视图等等。)

是否可以使用

之类的代码
persons.sum(_.age)

foldLeft / reduceLeft做什么?

2 个答案:

答案 0 :(得分:12)

你回答的是你自己。只需使用view

persons.view.map(_.age).sum

通过检查工作流程来说服自己:

persons.view.map { p =>
  println("invoking age")
  p.age
}.map { x =>
  println("modifing age")
  x + 0
}.sum

Vs的:

persons.map { p =>
  println("invoking age")
  p.age
}.map { x =>
  println("modifing age")
  x + 0
}.sum

答案 1 :(得分:11)

库中的方法sum不能以这种方式工作,但你可以编写自己的方法:

def mySum[T, Res](f: T => Res, seq: TraversableOnce[T])(implicit num: Numeric[Res]) = 
  seq.foldLeft(num.zero)((acc, b) => num.plus(acc, f(b)))

您还可以添加隐式转化,因此您可以将其称为seq.sum(f)而不是mySum(f, seq)(您可能需要使用与sum不同的名称来避免冲突):

case class SumTraversableOnce[T](val seq: TraversableOnce[T]) { 
  def sum[Res](f: T => Res)(implicit num: Numeric[Res]) = mySum(f, seq)(num) 
}

implicit def toSumTraversableOnce[T](seq: TraversableOnce[T]) = 
  SumTraversableOnce(seq)

或者,自Scala 2.10起,

implicit class SumTraversableOnce[T](val seq: TraversableOnce[T]) { 
  def sum[Res](f: T => Res)(implicit num: Numeric[Res]) = mySum(f, seq)(num) 
}