Question

我有一个数据帧列表，我想使用lapply对其应用自定义函数

这是我的功能：

  rename_cols_pattern <- function (df, pattern, replacement = "") {
  names(df) <- gsub(names(df), pattern = pattern, replacement = replacement)
}

如何对lapply使用此功能？这不起作用，因为缺少df变量。我如何传递df变量，该变量将是di_data列表中的数据框

di_data <- lapply(di_data, rename_cols_pattern(pattern = "X"))

我可以使它像这样工作：

di_data <- lapply(di_data, function(x) {
  names(x) <- gsub(names(x), pattern = "X", replacement = "")
  x
})

但是我希望功能分开并想了解如何实现

Answer 1

您可能错过了函数的return语句。

rename_cols_pattern <- function(df, pattern, replacement="") {
  names(df) <- gsub(names(df), pattern=pattern, replacement=replacement)
  return(df)
}

正常用法：

rename_cols_pattern(dat, pattern="X", replacement="COL")
#   COL1 COL2 COL3 COL4
# 1    1    4    7   10
# 2    2    5    8   11
# 3    3    6    9   12

使用lapply：

lapply(list(dat, dat), rename_cols_pattern, pattern="X", replacement="COL")
# [[1]]
#   COL1 COL2 COL3 COL4
# 1    1    4    7   10
# 2    2    5    8   11
# 3    3    6    9   12
# 
# [[2]]
#   COL1 COL2 COL3 COL4
# 1    1    4    7   10
# 2    2    5    8   11
# 3    3    6    9   12

数据：

dat <- structure(list(X1 = 1:3, X2 = 4:6, X3 = 7:9, X4 = 10:12), class = "data.frame", row.names = c(NA, 
-3L))

Answer 2

rename_with的创建是为了解决此类问题

library(tidyverse)

mtcars %>% 
  rename_with(.fn = ~ str_remove_all(.x,"X"))

使用lapply时如何将参数传递给用户定义的函数

2 个答案: