r - 如果为 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])