循环遍历 R 中最后一个 data.frame 的 data.frame 的变量

Question

我编写了一个函数，用于将 data.frame 中的任何一个变量（即列）更改为其唯一级别并返回更改后的 data.frame。

我想知道如何使用我的函数一次更改多个变量并获得一个包含所有更改的最终 data.frame？

我尝试了以下方法，但这会提供多个 data.frames 而只有 last data.frame 是 所需的输出：

data <- data.frame(sid = c(33,33, 41), pid = c('Bob', 'Bob', 'Jim'))

#== My function for ONE variable:
f <- function(data, what){
 data[[what]] <- as.numeric(factor(data[[what]], levels = unique(data[[what]])))
 return(data)
}  

# Looping over `what`:
what <- c('sid', 'pid')
lapply(seq_along(what), function(i) f(data, what[i]))

Answer 1

在函数中，我们可以将return改为data[[what]]

f <- function(data, what){
   data[[what]] <- as.numeric(factor(data[[what]], levels = unique(data[[what]])))
   data[[what]]
  }  

data[what] <- lapply(seq_along(what), function(i) f(data, what[i]))

或者做

data[what] <-  lapply(what, function(x) f(data, x))

或者干脆

data[what] <- lapply(what, f, data = data)