在 R 中添加列，其值基于其他列的变量“名称”

Question

如果我们有一个 data.frame ，请说类似

    ///// !col1!col2!col3
      ---------------
id123   1    0    0
      ---------------
!id435   0    1    0
      ---------------
!id777   0    0    1

我想创建一个新列 newcol，其中变量名称的值具有 '1'

需要数据

    ///// !col1!col2!col3!newcol
      ---------------------
id123   1    0    0   !col1
      ---------------------
!id435   0    1    0  !col2
      ---------------------
!id777   0    0    1  !col3

1) 有没有办法在 base 或 plyr 中做？ 2）（可选）如果 id123 在 col1 和 col2 中都有值 1 ，如何调整它？如何“添加”这些值，在 newcol 中用逗号分隔

temp$col1 <- c(1,0,0)
temp$col2 <- c(0,1,0)
temp$col3 <- c(0,0,1)

temp<-data.frame(temp$col1, temp$col2, temp$col3)

感谢您的支持 :)

Answer 1

我们可以在base R中使用max.col

temp$newcol <- names(temp)[max.col(temp, 'first')]

---

如果我们在同一行有多个 1，并且所有列名称为单个字符串

i1 <- which(temp2 ==1, arr.ind = TRUE)
temp2$newcol <- NA_character_
temp2$newcol[unique(i1[,1])] <-  tapply(names(temp2)[i1[,2]],
         i1[,1], FUN = toString)
temp2$newcol
#[1] "col1"       "col1, col2" "col3"     

这也将确保只分配给有 1 的行

数据

temp <- data.frame(col1  = c(1, 0, 0), col2 = c(0, 1, 0), col3 = c(0, 0, 1))
temp2 <- data.frame(col1 = c(1, 1, 0), col2 = c(0, 1, 0), col3 = c(0, 0, 1)) 

Answer 2

附加选项

library(tidyverse)
temp2 <- data.frame(col1 = c(1, 1, 0), col2 = c(0, 1, 0), col3 = c(0, 0, 1)) 

temp2 <- temp2 %>% 
  mutate(id = row_number())

temp2 %>% 
  pivot_longer(-id) %>% 
  filter(value == 1) %>% 
  group_by(id) %>% 
  summarise(col = str_c(name, collapse = ", ")) %>% 
  left_join(temp2) %>% 
  select(-id)