如何获取数据框中的行索引列表？

Question

我需要根据“第二列中的值== 1）规则获取行索引列表

例如，从此数据框中

   1 2
1  1 1
2  1 1
3  1 1
4  0 1
5  0 0
6  0 1

我需要一个列表

[1,2,3,4,6]

我刚从Python切换到R，在不同的数据类型中我有点迷茫。

Answer 1

您可以尝试

which(!!df$`2`) # applicable only if you only have values 0 and 1 in column 2

或

which(df$`2`==1) 

数据

df <- structure(list(`1` = c(1L, 1L, 1L, 0L, 0L, 0L), `2` = c(1L, 1L, 
1L, 1L, 0L, 1L)), class = "data.frame", row.names = c("1", "2", 
"3", "4", "5", "6"))