挑选具有正值的列

Question

我正在处理一个数据帧，并试图使用R来计算包含数据的列名。我有一个示例数据帧如下所示：

df <-structure(
  list(
    area = structure(1:9, 
      .Label = c("a", "b", "c", "d", "e", "f", "g", "h", "i"), 
       class = "factor"
    ), 
    cat = c(5L, 0L, 0L, 0L, 0L, 0L, 0L, 3L, 0L), 
    dog = c(0L, 10L, 0L, 0L, 0L, 0L, 0L, 0L, 11L), 
    rabbit = c(0L, 0L, 0L, 3L, 0L, 0L, 0L, 0L, 0L), 
    horse = c(0L, 0L, 8L, 0L, 0L, 0L, 0L, 0L, 0L), 
    sheep = c(0L, 0L, 0L, 0L, 9L, 0L, 0L, 0L, 0L), 
    mouse = c(0L, 0L, 0L, 0L, 0L, 5L, 0L, 0L, 0L), 
    rat = c(0L, 0L, 0L, 0L, 0L, 0L, 2L, 0L, 0L)
  ), 
  .Names = c("area", "cat", "dog", "rabbit", "horse", "sheep", "mouse", "rat"), 
  class = "data.frame", 
  row.names = c(NA, -9L)
)

在从“猫”到“老鼠”的列中，只有一列有正值（其余的是零）。我如何让R添加一个名为“动物”的额外列，给我每个区域的动物名称（从列名），其中有正值，即“动物”列将读作猫、狗、马、兔子、羊等？

我们将不胜感激地接受任何帮助或指导。

非常感谢。

Answer 1

df$animal <- names(df)[-1][apply( df[,-1] , 1, which.max )]
df
#  area cat dog rabbit horse sheep mouse rat animal
#1    a   5   0      0     0     0     0   0    cat
#2    b   0  10      0     0     0     0   0    dog
#3    c   0   0      0     8     0     0   0  horse
#4    d   0   0      3     0     0     0   0 rabbit
#5    e   0   0      0     0     9     0   0  sheep
#6    f   0   0      0     0     0     5   0    pos
#7    g   0   0      0     0     0     0   2    pos
#8    h   3   0      0     0     0     0   0    cat
#9    i   0  11      0     0     0     0   0    dog