考虑这个课程:
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
做什么?
答案 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)
}