如何使用dplyr粘贴列的子集

时间:2018-08-28 12:29:07

标签: r dplyr

如何使用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重写函数?

2 个答案:

答案 0 :(得分:4)

这是使用dplyrrlang的函数版本

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