r - 通过 R 中 foo(x, bar) 的第二个参数调用的通用函数

Question

我想为 R 中的“foo”类创建一个通用函数 predict()，以便在 predict() 的第二个参数属于“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