Scala中map调用多参数函数的实现方法咨询

嘿，这个问题其实很常见——Scala的map方法只接受单参数函数，而你现在把function改成了三参数，直接传肯定会报错。这里有几个简单好用的解决办法，我一个个给你拆解：

方法一：匿名函数包装（最直观的写法） #

你可以在map里写一个匿名函数，把列表中的元素作为第一个参数，再手动传入另外两个固定参数。代码示例如下：

def mapFunction(): StructType = {
  val elements: List[Int] = List(1, 2, 3)
  // 先定义你要传入的额外参数值（也可以是外部传入的变量）
  val anotherParam1 = 10
  val anotherParam2 = 20
  // 用匿名函数把多参数函数转换成map需要的单参数形式
  StructType(elements.map(listEle => function(listEle, anotherParam1, anotherParam2)))
}

方法二：偏应用函数（更简洁的写法） #

Scala支持用下划线_来表示“待填充的参数”，这样可以快速把多参数函数转换成单参数的偏应用函数，代码会更简洁：

def mapFunction(): StructType = {
  val elements: List[Int] = List(1, 2, 3)
  val anotherParam1 = 10
  val anotherParam2 = 20
  // 下划线代表第一个参数会由map自动传入
  StructType(elements.map(function(_, anotherParam1, anotherParam2)))
}

方法三：柯里化函数（如果允许修改function的定义） #

如果可以调整function的结构，把它改成柯里化形式，会让参数固定的逻辑更优雅：

// 柯里化后的function：把参数分成两组
def function(listEle: Int)(anotherParam1: Int, anotherParam2: Int): StructField = {
  // 你的业务逻辑不变
}

// 在mapFunction中调用
def mapFunction(): StructType = {
  val elements: List[Int] = List(1, 2, 3)
  val anotherParam1 = 10
  val anotherParam2 = 20
  // 先固定第二组参数，得到一个单参数函数
  val singleArgFunc = function(_: Int)(anotherParam1, anotherParam2)
  StructType(elements.map(singleArgFunc))
}

原理补充 #

本质上，List.map需要的是类型为Int => StructField的函数，而你的多参数function类型是(Int, Int, Int) => StructField。上面的所有方法都是在做类型适配——把多参数函数转换成map能接受的单参数函数形式。

内容的提问来源于stack exchange，提问作者John Fisherman