Scala中任意维度输入函数的偏应用泛化实现方法问询

当然可以实现，核心思路是将固定参数与剩余参数序列合并为完整的n维向量，再传递给原函数。以下是具体实现方案：

基础实现：普通函数版本 #

这个版本接受原函数、固定参数的位置i（从0开始计数）和参数值xi，返回一个新函数，该函数接受剩余的n-1个参数组成的序列，合并后调用原函数：

def partialApply(f: IndexedSeq[Double] => Double)(i: Int)(xi: Double): IndexedSeq[Double] => Double = {
  (remaining: IndexedSeq[Double]) =>
    // 将xi插入到剩余参数序列的第i位，生成完整n维向量
    val fullVector = remaining.take(i) ++ Seq(xi) ++ remaining.drop(i)
    f(fullVector)
}

测试示例（n=2的场景） #

// 定义n=2的原函数f(x,y)=sin(x+y)
val f: IndexedSeq[Double] => Double = seq => math.sin(seq(0) + seq(1))

// 固定第0位参数为2，得到仅接受y的函数
val fixedX = partialApply(f)(0)(2.0)
// 调用：传入y=3.0，计算sin(2+3)
println(fixedX(Seq(3.0))) // 输出sin(5)的近似值

// 固定第1位参数为2，得到仅接受x的函数
val fixedY = partialApply(f)(1)(2.0)
println(fixedY(Seq(3.0))) // 输出sin(3+2)的近似值

严谨版本：带参数校验 #

如果需要严格保证输入合法性，可以添加长度检查，避免原函数因向量长度错误抛出异常：

def partialApplyWithValidation(f: IndexedSeq[Double] => Double)(n: Int)(i: Int)(xi: Double): IndexedSeq[Double] => Double = {
  require(n > 0, "维度n必须大于0")
  require(i >= 0 && i < n, s"参数位置i必须在0到${n-1}之间")
  
  (remaining: IndexedSeq[Double]) =>
    require(remaining.length == n - 1, s"剩余参数序列长度必须为${n-1}")
    val fullVector = remaining.take(i) ++ Seq(xi) ++ remaining.drop(i)
    f(fullVector)
}

符合“偏函数”定义的版本 #

如果需要返回仅在合法输入域上有定义的偏函数（PartialFunction），可以用模式匹配实现：

def partialApplyAsPartialFunc(f: IndexedSeq[Double] => Double)(n: Int)(i: Int)(xi: Double): PartialFunction[IndexedSeq[Double], Double] = {
  require(n > 0, "维度n必须大于0")
  require(i >= 0 && i < n, s"参数位置i必须在0到${n-1}之间")
  
  // 仅当剩余参数序列长度为n-1时，函数才有定义
  case remaining if remaining.length == n - 1 =>
    val fullVector = remaining.take(i) ++ Seq(xi) ++ remaining.drop(i)
    f(fullVector)
}

偏函数的使用方式 #

val partialFunc = partialApplyAsPartialFunc(f)(2)(0)(2.0)

// 检查输入是否合法
println(partialFunc.isDefinedAt(Seq(3.0))) // true
println(partialFunc.isDefinedAt(Seq(3.0, 4.0))) // false

// 合法输入调用
println(partialFunc(Seq(3.0)))

与二元函数偏应用的关联 #

你原来实现的leftPartialFunction本质是这个泛化方案在n=2、i=0时的特例，泛化版本只是把固定单一位参数、合并序列的逻辑扩展到了任意维度和任意参数位置。

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