将列名和数据框传递给R中的自定义函数

Question

我正在尝试在 R 中编写一个函数：

1) 接收数据框和列名作为参数。 2) 对数据框中的列执行操作。

func <- function(col, df)
{
  col = deparse(substitute(col))
  print(paste("Levels: ", levels(df[[col]])))
}


func(Col1, DF)

func(Col2, DF)

mapply(func, colnames(DF)[1:2], DF)

输出

> func(Col1, DF)
[1] "Levels:  GREEN"  "Levels:  YELLOW"

> func(Col2, DF)
[1] "Levels:  0.1" "Levels:  1"  

> mapply(func, colnames(DF)[1:2], DF)
 Error in `[[.default`(df, col) : subscript out of bounds 

Answer 1

两件事：

• 在您的函数func 中，您将deparse(substitute(col)) 应用于您期望的对象col 不是字符串。所以它适用于func(Col1, DF)。但是在您的mapply() 调用中，您的参数colnames(...) 是一个字符串，因此会产生错误。使用func('Col1', DF) 获得相同的错误。

• 在mapply() 调用中，所有参数都必须是向量或列表。所以你需要使用list(df, df)，或者如果你不想复制，去掉你的函数func的参数df。

这是一种可行的替代方法：

func <- function(col, df)
{
  print(paste("Levels: ", levels(df[,col])))
}

mapply(FUN = func, colnames(DF)[1:2], list(DF, DF))

Answer 2

请查看@demarsylvain 的最后一条评论 - 可能是您这边的复制粘贴错误，您应该这样做：

func <- function(col,df) {
  print(paste("Levels: ", levels(df[,col])))
}

mapply(FUN = func, c('Species', 'Species'), list(iris, iris))

你做到了：

func <- function(col) {
  print(paste("Levels: ", levels(df[,col])))
}

mapply(FUN = func, c('Species', 'Species'), list(iris, iris))

请投票并接受@demarsylvain 的解决方案，它有效

编辑以解决您的评论：

要获得任意列名列表的通用版本，您可以使用此代码，抱歉循环:)

func <- function(col,df) {
  print(paste("Levels: ", levels(df[,col])))
}

cnames = colnames(iris)


i <- 1
l = list()
while(i <= length(cnames)) {
  l[[i]] <- iris
  i <- i + 1
}

mapply(FUN = func, cnames, l)