19

对于任何特定类型的数字（即双/整数）来说，这似乎很容易出现问题，但在一般情况下很难编写。

implicit def iterebleWithAvg(data:Iterable[Double]) = new {
    def avg:Double = data.sum / data.size
}

如何为任何类型的数字（Int，Float，Double，BigDecemial）实现这一点？

4

2 回答 2

33

您必须传递一个隐式Numeric，这将允许求和和转换为 Double：

def average[T]( ts: Iterable[T] )( implicit num: Numeric[T] ) = {
  num.toDouble( ts.sum ) / ts.size
}

编译器将为您提供正确的实例：

scala> average( List( 1,2,3,4) )
res8: Double = 2.5

scala> average( 0.1 to 1.1 by 0.05 )
res9: Double = 0.6000000000000001

scala> average( Set( BigInt(120), BigInt(1200) ) )
res10: Double = 660.0

您可以使用该函数来定义隐式视图（前提是您传播隐式数字依赖项）：

implicit def iterebleWithAvg[T:Numeric](data:Iterable[T]) = new {
  def avg = average(data)
}

scala> List(1,2,3,4).avg
res13: Double = 2.5
于 2012-04-15T07:50:42.323 回答
14

这是我在代码中定义它的方式。

Numeric我没有使用 ，而是使用Fractional，因为Fractional定义了除法运算（Numeric不一定有除法）。这意味着当你调用时.avg，你会得到你输入的相同类型，而不是总是得到Double。

我还在所有GenTraversableOnce集合中定义它，以便它适用于例如Iterator.

class EnrichedAvgFractional[A](self: GenTraversableOnce[A]) {
  def avg(implicit num: Fractional[A]) = {
    val (total, count) = self.toIterator.foldLeft((num.zero, num.zero)) {
      case ((total, count), x) => (num.plus(total, x), num.plus(count, num.one))
    }
    num.div(total, count)
  }
}
implicit def enrichAvgFractional[A: Fractional](self: GenTraversableOnce[A]) = new EnrichedAvgFractional(self)

请注意，如果我们给它一个集合Double，我们会返回Double，如果我们给它BigDecimal，我们会返回BigDecimal。我们甚至可以定义自己的Fractional数字类型（我偶尔会这样做），它会为此工作。

scala> Iterator(1.0, 2.0, 3.0, 4.0, 5.0).avg
res0: Double = 3.0

scala> Iterator(1.0, 2.0, 3.0, 4.0, 5.0).map(BigDecimal(_)).avg
res1: scala.math.BigDecimal = 3.0

然而，Intis not a kind of ，这意味着获得 an和 s 的平均结果是Fractional没有意义的，所以我们必须有一个特殊情况来转换为 a 。IntIntIntDouble

class EnrichedAvgInt(self: GenTraversableOnce[Int]) {
  def avg = {
    val (total, count) = self.toIterator.foldLeft(0, 0) {
      case ((total, count), x) => (total + x, count + 1)
    }
    total.toDouble / count
  }
}
implicit def enrichAvgInt(self: GenTraversableOnce[Int]) = new EnrichedAvgInt(self)

所以平均Ints 给我们一个Double：

scala> Iterator(1, 2, 3, 4, 5).avg
res2: Double = 3
于 2012-04-15T16:36:07.297 回答