r - 如何创建新列并通过 r 中的选定行添加列名

Question

a<-c(TRUE,FALSE,TRUE,FALSE,TRUE,FALSE)
b<-c(TRUE,FALSE,TRUE,FALSE,FALSE,FALSE)
c<-c(TRUE,TRUE,TRUE,FALSE,TRUE,FALSE)
costumer<-c("one","two","three","four","five","six")
df<-data.frame(costumer,a,b,c)

这是一个示例代码。看起来像这样打印:

  costumer     a     b     c 
1      one  TRUE  TRUE  TRUE  
2      two FALSE FALSE  TRUE 
3    three  TRUE  TRUE  TRUE 
4     four FALSE FALSE FALSE 
5     five  TRUE FALSE  TRUE 
6      six FALSE FALSE FALSE

我想创建一个新列 df$items，它只包含数据中每一行的 TRUE 列名。像这样:

  costumer     a     b     c items
1      one  TRUE  TRUE  TRUE  a,b,c
2      two FALSE FALSE  TRUE  c
3    three  TRUE  TRUE  TRUE  a,b,c
4     four FALSE FALSE FALSE 
5     five  TRUE FALSE  TRUE 
6      six FALSE FALSE FALSE

我想到了使用 apply 函数或使用 which 来选择索引，但无法弄清楚。谁能帮帮我？

Answer 1

df$items <- apply(df, 1, function(x) paste0(names(df)[x == TRUE], collapse = ","))
df
  custumer     a     b     c items
1      one  TRUE  TRUE  TRUE a,b,c
2      two FALSE FALSE  TRUE     c
3    three  TRUE  TRUE  TRUE a,b,c
4     four FALSE FALSE FALSE      
5     five  TRUE FALSE  TRUE   a,c
6      six FALSE FALSE FALSE