Question

我正在编写一个函数，我希望能够将向量和公式作为第一个参数。如果是矢量，我做一些单变量计算，如果是公式，我用第二个变量分析第一个变量（第二个变量总是一个因子）。

这是我目前的代码：

fun = function(formula,data) {

  if (class(with(data,formula))=="formula") {
    mod = model.frame(formula,data)
    n.group=names(mod)[2] 
    group <- eval(parse(text=paste("mod$",n.group,sep=""))) #x
    response <- model.response(mod) # y
    return(table(response,group))
  }

  else {
    return(table(with(data,formula)))
  }
}

data(iris)

fun(Sepal.Length~Species,iris) # works correctly
fun(Sepal.Length,iris) # returns an error

返回值仅用于说明。

干杯！

Answer 1

试试这个：

fun.formula <- function(formula, data) {
  mod = model.frame(formula, data)
  n.group <- names(mod)[2] 
  group <- eval(parse(text=paste("mod$",n.group,sep=""))) #x
  response <- model.response(mod) # y
  table(response, group)
}

fun <- function(formula, data) {
    ret <- try( table(eval(substitute(formula), data), silent = TRUE)
    if (inherits(try, "try-error)) fun.formula(formula, data) else ret
}

# tests
fun(Sepal.Length ~ Species, iris)
fun(Sepal.Length, iris)

也就是说，这是一个相当不寻常的接口，相反，最好通过将其名称作为字符串来指定公式为变量的情况，在这种情况下可以实现更常见的S3实现：

fun2 <- function(formula, data, ...) UseMethod("fun2")
fun2.formula <- fun.formula
fun2.character <- function(formula, data) table(data[[formula]])

# tests
fun2(Sepal.Length ~ Species, iris)
fun2("Sepal.Length", iris) # with this approach use a character string

已修订现在我们使用try并添加了S3方法。

Answer 2

理想情况下，我会使用S3方法解决这个问题，但我无法弄清楚如何做到这一点。以下完成了工作：

fun <- function(x,data) {
  mod = try(model.frame(x,data),silent=T)
  if (inherits(mod, "try-error")) {
    x=data[,deparse(substitute(x))]
    return(table(x))
  }
  else {
    mod = model.frame(x,data)
    n.group=names(mod)[2] 
    group <- eval(parse(text=paste("mod$",n.group,sep=""))) #x
    response <- model.response(mod) # y
    return(table(response,group))    
  }
}

fun(Sepal.Length~Species,iris) # works correctly
fun(Sepal.Length,iris) # works!

如何编写可以将向量或公式作为第一个参数的R函数？

2 个答案: