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



我从这个网站(英国国家统计局(下载了一个.ods文件。由于表单的结构方式,我将其作为两个独立的数据帧导入:

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 pointnTotal income before tax',
'2008-09' = '2008-09(a)')
income_pretax['2008-09'] <- as.numeric(income_pretax$'2008-09')

不过,我正在努力将以上内容转化为一个函数。我认为它应该是类似下面的东西,但老实说,我不知道如何告诉R我正在向函数传递多个数据帧,也不知道如何处理多个变量有人能对此提出建议吗

##Attempting a function
cleanvars <- function(data, varlist){
data <- data %>% 
rename(pp_tot_income_pretax = {{varlist}})

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

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

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

我们可以在base R中做到这一点

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

最新更新