将数据操作转换为 R 中的函数

Question

我从 this website（英国国家统计局）下载了一个 .ods 文件。由于 sheet 的结构方式，我将其导入为两个单独的数据帧：

library(readODS)
income_pretax <- read_ods('/Users/c.robin/Downloads/NS_Table_3_1a_1819.ods', range = "A4:U103")
income_posttax <- read_ods('/Users/c.robin/Downloads/NS_Table_3_1a_1819.ods', range = "A104:U203")

我想对两个数据帧进行一些清理：更改两个变量的名称并将其中一个变量重铸为数字。这就是我所拥有的，它适用于单个 df:

income_pretax <- income_pretax %>% 
  rename(pp_tot_income_pretax = 'Percentile point\nTotal income before tax',
         '2008-09' = '2008-09(a)')

income_pretax['2008-09'] <- as.numeric(income_pretax$'2008-09')

不过，我正在努力将上述内容转化为函数。我认为它应该像下面这样 something，但老实说我不知道如何告诉 R 我正在将多个数据帧传递给函数，也不知道如何处理多个变量。 任何人都可以对此提出建议吗？

##Attempting a function
cleanvars <- function(data, varlist){
  data <- data %>% 
    rename(pp_tot_income_pretax = {{varlist}})
  
  data['2008-09'] <- as.numeric(data$'2008-09')
}

Answer 1

您可以将命名向量传递给函数。

library(dplyr)

cleanvars <- function(data, varlist){
   data %>% rename(varlist)
}

cleanvars(mtcars %>% head, c('new_mpg' = 'mpg', 'new_cyl' = 'cyl'))

#                  new_mpg new_cyl disp  hp drat    wt  qsec vs am gear carb
#Mazda RX4            21.0       6  160 110 3.90 2.620 16.46  0  1    4    4
#Mazda RX4 Wag        21.0       6  160 110 3.90 2.875 17.02  0  1    4    4
#Datsun 710           22.8       4  108  93 3.85 2.320 18.61  1  1    4    1
#Hornet 4 Drive       21.4       6  258 110 3.08 3.215 19.44  1  0    3    1
#Hornet Sportabout    18.7       8  360 175 3.15 3.440 17.02  0  0    3    2
#Valiant              18.1       6  225 105 2.76 3.460 20.22  1  0    3    1

Answer 2

我们可以在 base R

nm1 <- c('mpg', 'cyl')
nm2 <- paste0("new_", nm1)
i1 <- match(nm1, names(mtcars))
names(mtcars)[i1] <- nm2

将数据操作转换为 R 中的函数

Turning data manipulation into a function in R

r

rename

function

dplyr