我想编写一个函数,它接受任何用户提供的数学函数(例如,x ^ 2)并使用它做不同的事情,例如:
#-----------------nonworking code---------------------
foo <- function(FUN, var){
math_fun <- function(x){
FUN
}
curve(math_fun, -5, 5) #plot the mathematical function
y = math_func(var) #compute the function based on a user provided x value.
points(x=var, y=y) #plot the value from the last step.
}
#A user can use the function defined above in a way as shown below:
Function <- x^2 + x
foo(FUN=Function, var = 2)
但显然这个功能不起作用:
首先,如果我运行此函数,我会得到Error in math_fun(x) : object 'x' not found
。
其次,即使函数确实有效,我假设变量是x
,但用户可以使用任何字母。
对于第二个问题,一个可能的解决方案是要求用户指定他们用作变量的字母。
foo <- function(FUN, var, variable){
math_fun <- function(variable){
FUN
}
curve(math_fun, -5, 5)
y = math_func(var)
points(x=var, y=y)
}
但是我对于如何实现这一点感到很遗憾......如果有人可以帮我解决至少部分问题,那就太棒了。谢谢!
答案 0 :(得分:6)
比这简单得多。用户定义的函数应该在其定义中包含参数,例如function(x) x^2 + x
代替x^2 + x
。然后它可以直接传递和调用:
foo <- function(math_fun, var){
curve(math_fun, -5, 5) #plot the mathematical function
y = math_fun(var) #compute the function based on a user provided x value
points(x=var, y=y) #plot the value from the last step.
}
#A user can use the function defined above in a way as shown below:
Function <- function(x) x^2 + x
foo(Function, var = 2)