如果将列转换为以逗号分隔的列表1

Question

我正在尝试将包含1或0的某些列转换为带有列名称的逗号分隔列表。

更具体地，举一个例子：

df <- data.frame(id = c(1, 2, 3, 4), A = c("0", "1", "1", "0"), B = c("1", "1", "0", "0"), C = c("0", "0", "1", "1"))

如果列A，B和C等于1，则应创建一个新的“列表”列，其中包含所有逗号分隔的“标记”（列名）。

id  A   B   C
1   0   1   0
2   1   1   0
3   1   0   1
4   0   0   1

所需结果：

id  A   B   C   list
1   0   1   0   B
2   1   1   0   A, B
3   1   0   1   A, C
4   0   0   1   C

Answer 1

这是apply来自的选项base R

df$new <- apply(df[-1], 1, function(x) toString(names(x)[x==1]))

在这里，我们正在创建一个字符串列，但是如果要创建一个list列

df$new <- apply(df[-1], 1, function(x) names(x)[x==1])