如何在scala中设置函数参数的默认值

Question

我正在尝试为以下函数中的参数 keyFunction 设置默认值 (x) => x：

def count[A, B](list: List[A], keyFunction: (A) => B, isRatio : Boolean = false): Map[B, Double] = {
    lazy val number = list.size.toDouble
    list.groupBy(keyFunction).map{
      case (key, group) => if (isRatio) (key, group.size/number) else (key, group.size.toDouble)
    }
  }

我知道一个愚蠢的解决方案就是定义另一个函数：

  def simplyCount[A] (list: List[A], isRatio: Boolean = false): Map[A, Double] = count(list, (e: A) => e, isRatio)

我怎样才能更好地做到这一点？提前致谢。

Answer 1

您可以将默认参数添加为 (x: A) => x，但如果您希望 Scala 仅在给定 list 的情况下自动推断类型，则必须将其分隔到单独的参数列表中：

def count[A, B](list: List[A])(keyFunction: (A) => B = (x: A) => x, isRatio: Boolean = false): Map[B, Double] {
  ???
}

count(List("a", "B", "a", "c"))()
// Map(a -> 2.0, c -> 1.0, B -> 1.0)

您可以将它作为一个参数列表来完成，但是您必须明确地为其指定类型，因为类型推断将无法计算出来：

def count[A, B](list: List[A], keyFunction: (A) => B = (x: A) => x, isRatio: Boolean = false): Map[B, Double] {
  ???
}

count[String,String](List("a", "B", "a", "c"))
// Map(a -> 2.0, c -> 1.0, B -> 1.0)

如何在scala中设置函数参数的默认值

how to set default values for function parameters in scala

scala

functional-programming

type-parameter