在R中使用foo（x，bar）的第二个参数调用的泛型函数R

Question

我想为R中的'foo'类创建一个通用函数predict（），以便当predict（）的第二个参数是class时调用它'foo'：

class(y.foo) <- "foo" 

predict(x, y.foo) 

可能吗？怎么做？

Answer 1

试试这个。这里我们foo方法只是输出"foo"和真正的代码，它将与任何foo方法来代替您有：

predict <- function(x, foo = NULL, ...) UseMethod("predict", foo) 
predict.foo <- function(x, foo, ...) "foo" # replace with your foo method 
predict.default <- function(x, foo = NULL, ...) if (is.null(foo)) 
     stats::predict(x, ...) else stats::predict(x, foo, ...) 

现在测试一下：

y.foo <- 1 
class(y.foo) <- "foo" 
predict(0, y.foo) 
## [1] "foo" 

fm <- lm(demand ~ Time, BOD) 

predict(fm) 
##  1  2  3  4  5  6 
## 10.24286 11.96429 13.68571 15.40714 17.12857 20.57143 

predict(fm, newdata = list(Time = 1:2)) 
##  1  2 
## 10.24286 11.96429 

predict(fm, list(Time = 1:2)) # same 
##  1  2 
## 10.24286 11.96429