如何使用dplyr将data.frame的动态列子集粘贴在一起?
如果我具有以下功能:
d = data.frame(a = 1:5, b=2:6, c=3:7)
fun = function(d, colnames){
d$f = Reduce(function(x,y)paste(x,y,sep='_'), d[,colnames])
return(d)
}
fun(d, c('a','b'))
fun(d, c('b','c'))
fun(d, c('a','b','c'))
如何使用dplyr重写函数?
答案 0 :(得分:4)
这是使用dplyr
和rlang
的函数版本
library(dplyr)
library(rlang)
fun = function(d,a,b){
a.var <- enquo(a)
b.var <- enquo(b)
d %>% mutate(f = paste(!!a.var,!!b.var, sep="_"))
}
fun(d,a,b)
fun = function(d,...){
#browser()
vars <- quos(...)
d %>% mutate(f = paste(!!!vars, sep="_"))
}
fun(d,a)
fun(d,c,a)
fun(d,a,b,c)
答案 1 :(得分:1)
这适用于您的示例:
> library(dplyr)
> d = data.frame(a = 1:5, b=2:6, c=3:7)
> d <- d %>% mutate(f = paste(a,b,sep='_'))
> d
a b c f
1 1 2 3 1_2
2 2 3 4 2_3
3 3 4 5 3_4
4 4 5 6 4_5
5 5 6 7 5_6
编辑:
编辑后,它将对您有效:
> library(dplyr)
> d = data.frame(a = 1:5, b=2:6, c=3:7)
> colnames <- c("a", "c")
> d <- d %>% mutate(f = paste(!!as.name(colnames[1]),!!as.name(colnames[2]),sep='_'))
> d
a b c f
1 1 2 3 1_3
2 2 3 4 2_4
3 3 4 5 3_5
4 4 5 6 4_6
5 5 6 7 5_7