0

我编写了一个对单个向量进行操作的函数。有时我想整体使用该功能data.frame。我可以通过使用sapply相关变量来实现这一点,但现在我想使用 S3 方法包含和指导函数。

首先,设置:

df_test <- data.frame(x = c(1, 2, 3, 4, 5), 
                      y = c(6, 7, 8, 9, 10), 
                      z = c(11, 12, 13, 14, 15))

adder <- function(x, from, to) {
  if(is.null(ncol(x))) {
    x <- structure(list(x = x, from = from, to = to), class = "single")
  } else {
    x <- structure(list(x = x, from = from, to = to), class = "multiple")
  }
  UseMethod("adder", x)
}

adder.single <- function(x, from, to) {
  x <- x + (from - to) # just some random operation here - the real function does more
  x
}

adder.multiple <- function(x, from, to) {
  x <- sapply(x, function(y) { 
    y <- structure(list(x = y, from = from, to = to), class = "single"); 
    UseMethod("adder", y) })
  x
}

因此,在单个向量上,该函数起作用:

> adder(df_test[,1], 2, 4)
[1] -1  0  1  2  3

但是通过整个data.frame不会:

> adder(df_test, 2, 4)
Error in from - to : 'from' is missing

我很清楚问题是什么 -adder检测到我们正在查看一个整体data.frame,使用“multiple”方法,然后调用“single”方法和参数from并且to没有被传递。

我阅读了 Nicolas Christian 的 Hadley Wickham 的OOP Field GuideAdvanced Programming,但对我来说并没有什么用。我欢迎完全不同的方法,只要它使用 S3 方法,因为这个练习的一部分是让我学习如何使用它们。

4

1 回答 1

2

只需调用adder,S3 调度就会做正确的事情并调用正确的函数。

adder.multiple <- function(x, from, to) {
  sapply(x,  adder,from,to)
}

你不应该定义泛型adder两次(调用UseMethod)。

于 2014-01-24T20:09:28.827 回答