scala способ определения функций, принимающих список различных числовых типов

У меня следующая проблема: у меня есть функция, которая принимает List [Double] в качестве параметра, выполняет некоторые арифметические операции с элементами списка и затем возвращает результат. Я бы хотел, чтобы функция также принимала List [Int]. Вот пример:

def f(l: List[Double]) = {
    var s = 0.0 
    for (i <- l)
        s += i
    s
}

val l1 = List(1.0, 2.0, 3.0)
val l2 = List(1, 2, 3)

println(f(l1))
println(f(l2))

Конечно, второй println не работает, поскольку f требует List [Double], а не List [Int].

Также обратите внимание на формулировку суммы в функции f не в стиле Scala, чтобы доказать необходимость использования 0 (или других констант) внутри самой функции (если я суммирую значения Int, мне нужно инициализировать s на 0, а не на 0,0.

Каков наилучший способ (меньше кода) заставить функцию работать как с Double, так и с Int?

(Я видел кое-что о 2.8 Numeric trait, я не совсем уверен, как его использовать ...)

Спасибо всем за помощь.


scala function-parameter scala-2.8
person Filippo Tabusso    schedule 10.02.2010    source источник

Ответы (2)


arrow_upward
7
arrow_downward

В scala 2.8 и использовании числового комбинирования для неявного преобразования ваш пример может быть записан как:

import Numeric._
def f[T](l: List[T])(implicit n: Numeric[T]):T = {
    var s = n.zero
    for (i <- l)
        s = n.plus(s, i)
    s
}

val l1 = List(1.0, 2.0, 3.0)
val l2 = List(1, 2, 3)

println(f(l1))
println(f(l2))

//or
def f2[T](l: List[T])(implicit n: Numeric[T]):T = {
 import n._
 var s = zero
 for (i <- l)
   s += i
 s
}
println(f2(l1))
println(f2(l2))

А теперь еще один пример выполнения суммы более масштабным способом:

def sum[T](l:List[T])(implicit n: Numeric[T]):T = {
 import n._
 l.foldLeft(zero)(_ + _)
}

println(sum(l1))
println(sum(l2))

//or since 2.8 Seq include already a sum function
def sum[T](l:List[T])(implicit n: Numeric[T]):T = l.sum

println(sum(l1))
println(sum(l2))
person Patrick    schedule 10.02.2010
comment
def sum[T : Numeric](l:List[T]) = l.sum подойдет. - person Thomas Jung; 10.02.2010

arrow_upward
4
arrow_downward

В этом ответе используется черта Numeric.

import Numeric._
def f[A](l: List[A])(implicit numeric: Numeric[A]) = 
  l reduceLeft ((l,r) => numeric.plus(l, r))

Или используя границы контекста:

def f[A : Numeric](l: List[A]) =
   l.reduceLeft((l,r) => implicitly[Numeric[A]].plus(l, r))
person Thomas Jung    schedule 10.02.2010