Question

我有一个函数，它将另一个函数作为输入变量，例如

WrapperFunction <- function(x, BaseFunction){
   y <- cor(x)
   BaseFunction(y)
}

现在我想在这个WrapperFunction中输入各种BaseFunction，以生成输出向量，但是如何定义一个函数列表，以便不用手动插入每个函数，我可以使用for循环自动化该过程：

for (i in 1:n){
   output[i] <- WrapperFunction(x, FunctionList[i])
}

我尝试过定义

FunctionList <- list()

FunctionList[1] = Function1 , etc....

，这不起作用。

也没有定义

FunctionList <- c("Function1", "Function2", ...)

Answer 1

我不确定您使用的x值是什么，我认为它是一个矩阵。您可以将一些函数放入列表中，然后使用lapply。

例如

m <- matrix(1:6, 2)
lst <- list(cov, diff)
lapply(lst, WrapperFunction, x = m)
#[[1]]
#              [,1]          [,2]          [,3]
#[1,]  1.848893e-32 -6.162976e-33 -6.162976e-33
#[2,] -6.162976e-33  1.848893e-32 -6.162976e-33
#[3,] -6.162976e-33 -6.162976e-33  1.848893e-32
#
#[[2]]
#              [,1]          [,2]         [,3]
#[1,] -2.220446e-16  2.220446e-16 0.000000e+00
#[2,]  0.000000e+00 -2.220446e-16 2.220446e-16

另外，正如@nrussell所建议的那样，你可以对BaseFunction参数进行矢量化，这与上面做的基本相同。

vWF <- Vectorize(WrapperFunction, vectorize.args = "BaseFunction")
vWF(m, lst)

这会产生与上面相同的结果。

Answer 2

如果您想拥有一系列功能，可以执行以下操作：

myFuns <- list(mean, sd)

然后您可以lapply覆盖此列表，或者根据需要使用for循环。如果使用for循环，请确保使用[[语法，因为这样可以确保您正在检索函数而不是长度列表：

for (i in 1:n){
    output[i] <- WrapperFunction(x, myFuns[[i]])
}

或

lapply(myFuns, WrapperFunction, x = x)

Answer 3

略有不同的方式，希望这能回答你的问题...

  ## Function 1
  fun1 <-function(a)
  {
    a = a+a
    return(a)
  }

  ## Function 2
  fun2 <- function(a)
  {
    a = a*a
    return(a)
  }

  ## Function 3
  fun3 <- function(a)
  {
  a = a^a
  return(a)
  }

  ## Vector of function names
  funlist <- c("fun1","fun2","fun3")

  ## Main Wrapper fnction
  mainfun <- function(x,funlist)
  {
    y = cor(x[,1],x[,2])

    res=NULL
    for(i in 1:length(funlist))
    {
      k <- eval(parse(text=paste(funlist[i],"(a=",y,")",sep="")),env=.GlobalEnv)
      res <- c(res,k)
    }

    return(c(y,res))
  }

  #############

  x=cbind(c(1:100),c(101:200))
  mainfun(x,funlist)

 [1] 1 2 1 1

如何在R中定义函数列表/向量？

3 个答案: