r - 在 R 的 which() 和 ifelse() 函数中组合逻辑语句

Question

我总是使用这样的命令:

which(foo$bar == 'A' | foo$bar == 'B' | foo$bar == 'C')

由于它们都与同一个变量相关，我希望能够清理我的代码并执行以下操作:

which(foo$bar == 'A|B|C')  # such syntax works in grep, why not here?
# or...
which(foo$bar == c('A', 'B', 'C'))

但这都不起作用!我很确定一定有一个简单的解决方案，我就是找不到。我在 ifelse() 也有同样的问题功能，因此可以为通用解决方案提供额外的吹嘘权。

Answer 1

with(foo, which(bar %in% LETTERS[1:3]) )

可用于从数据框中选择行。也可以将其用作报告向量的逻辑索引，尽管使用逻辑索引，您确实需要记住 R 索引不是基于 0 的。:

  set.seed=(123)
  foo <- data.frame(bar=sample(LETTERS[1:15], 10))
  c("Not in A|B|C", "In A|B|C") [ 1+ foo$bar %in% LETTERS[1:3] ]