Question

我试图使用通用函数对数据框的每一列进行操作，其中操作将根据列的类别而变化。

我无法让函数访问列的名称，同时还将列调度到正确的方法。

df <- data.frame(f1 = factor(rep(1:3, 2)))

myfun <- function(x){
  UseMethod("myfun", x)
}

myfun.factor <- function(x){
  print("Using factor method")
  print(names(x))
  print(class(x))
}

myfun.default <- function(x){
  print("Using default method")
  print(names(x))
  print(class(x))
}

作为列表应用会提供正确的分派，但会从列中删除名称

library(plyr)
l_ply(df, myfun)
[1] "Using factor method"
NULL
[1] "factor"

作为数组应用会保留名称，但不会给出正确的名称

a_ply(df, 2, myfun)
[1] "Using default method"
[1] "f1"
[1] "data.frame"

是否有一种巧妙的方法可以充分利用这两者或am I stuck with the method described in the answer to this question?

Answer 1

我无法找到这个问题的直接解决方案，但是我确实找到了一个我非常满意的工作。我使用了一个中间函数，它通过a*ply传递给它的列，并将unlist ed形式及其名称传递给泛型函数。

myfun2 <- function(x, x_name){
  UseMethod("myfun2", x)
}

myfun2.factor <- function(x, x_name){
  print("Using factor method")
  print(x_name)
  print(class(x))
  return(NULL)
}

myfun2.default <- function(x, x_name){
  print("Using default method")
  print(x_name)
  print(class(x))
  return(NULL)
}

dispatch_fun <- function(x){
  myfun2(unlist(x), names(x))
}

a_ply(df, 2, dispatch_fun)
## [1] "Using factor method"
## [1] "f1"
## [1] "factor"

使用plyr和泛型函数对数据框的每一列进行操作，同时保留名称

1 个答案: