Question

我为模糊的标题表示歉意，想不出任何简短而具体的内容。

我有一个函数foo，它执行一些无关的计算。我已经采用以下方法。

foo.data.frame <- function(data, col1, col2){
  return(data.frame(col1 = data[[col1]],
                    col2 = data[[col2]]))
}

foo.list <- function(data){
  z <- unlist(data)
  d <- names(z)
  return(data.frame(col1 = z, col2 = d))
}

foo.character <- function(data){ #here data is a file address
  if(file.exists(data)){
    return(read.csv(data, row.names = 1))
  }
}

一些数据：

df <- data.frame(col1 = letters[1:10],
                 col2 = 1:10)

lis <- as.list(df$col2)
names(lis) <- df$col1

write.csv(df, "df.csv")

这些功能执行：

foo.data.frame(df, "col1", "col2"),
foo.list(lis)
foo.character("df.csv")

我正在努力创建一种将向量作为col1和col2的输入的方法：

foo.??? <- (data = NULL, col1, col2){
  return(data.frame(col1, col2))
  }

将被这样使用：

foo(col1 = df$col1, col2 = df$col2)

可以是

foo.default

还是不

我应该怎么做？

请注意，使用“可怜的男人S3”（当所有这些功能都在一个内部时，并使用if-then-else子句将适当的类的对象定向到适当的方法）是很容易做到的。

Answer 1

您将需要使用foo泛型来处理。例如，

foo.missing <- function(col1, col2) {
  data.frame(col1 = col1, col2 = col2)
}

foo <- function(data, ...) {
  if (missing(data) || is.null(data)) foo.missing(...)
  else UseMethod("foo")
}

# test    
foo(col1 = df$col1, col2 = df$col2)
foo(df, "col1", "col2"),
foo(lis)
foo("df.csv")

S3方法函数参数-类灵活性

1 个答案: