如何在 R 中创建一个函数来创建列的子集？

Question

我在 R 中有一个数据框，列中有将近 40 年（从 1980 年到 2019 年），我需要一个函数来创建几个采用相同变量并重命名的子集，然后我必须在下面放入其他数据框.

X1980 <- subset(all_data,select=c("Pais","RubroId","X1980"))
names(X1980)[names(X1980) == 'X1980'] <- 'Valor'
X1980$ANIO <- 1980

X1981 <- subset(all_data,select=c("Pais","RubroId","X1981"))
names(X1981)[names(X1981) == 'X1981'] <- 'Valor'
X1981$ANIO <- 1981

X1982 <- subset(all_data,select=c("Pais","RubroId","X1982"))
names(X1982)[names(X1982) == 'X1982'] <- 'Valor'
X1982$ANIO <- 1982

final_data <- rbind(X1980,X1981,X1982)

Answer 1

我们可以创建一个函数，其中列 select (cols_select) 作为字符向量，names_to_change（'Valor' 或任何其他列名称），子集通过 selecting 列的数据集 ('dat')，设置列名并创建一个新列 'ANIO'

f1 <- function(dat, cols_select, names_to_change){
     yearcol <- grep("^X\d{4}$", cols_select, value = TRUE)
     tmpdat <- subset(dat,select=cols_select)
     names(tmpdat)[names(tmpdat) == yearcol] <- names_to_change
     tmpdat$ANIO <- as.integer(sub("^X", "", yearcol))
     tmpdat
  }

并在 lapply

中使用它

nm1 <- paste0("X", 1980:1982)
out <- do.call(rbind, lapply(nm1, function(x)
         f1(dat, cols_select = c("Pais","RubroId", x), "Valor")))

如何在 R 中创建一个函数来创建列的子集？

How can I make a function in R to create subsets of columns?

r

dataframe

subset

rename

recycle