r - 根据数据框子集中的条件在 R 中添加新列

Question

我有以下数据集：

 df <- data.frame("attribute"=c('name', 'age', 'location'),
                "A"=c(1,0,0), "B"=c(0,1,1), "C"=c(1,0,1))
 names(df) <- c("attribute", "A", "B", "C")
 df

    attribute   A B C
        name    1 0 1
        age     0 1 0
     location   0 1 1

• 我想在 R 中添加一个新列，其中包含 A、B C 中的列名之一。
• 条件是属性之一name, age, location应与 A、B C 列之一匹配。

• 属性值为 one 的第一个列名称应作为附加列列出，如下所示：

    attribute   A B C Column
        name    1 0 1  A
        age     0 1 0  B
     location   0 1 1  B
  

  for(i in df$attribute){for(j in c("A", "B", "C")}print(names(df) [names(df)%in%c("A", "B ", "C")][i]==1)

我无法解决这个问题。

标签： rloopsdataframe

Answer 1

我们可以用max.col

df$Column <- names(df)[-1][max.col(df[-1], 'first')]
df$Column
#[1] "A" "B" "B"