是否有 R 函数用于查找向量中元素的索引？

Question

在 R 中，我有一个元素 x 和一个向量 v。我想找到 v 中等于 x 的元素的第一个索引。我知道执行此操作的一种方法是：which(x == v)[[1]]，但这似乎效率很低。有没有更直接的方法来做到这一点？

对于奖励积分，如果 x 是向量，是否有一个函数可以工作？也就是说，它应该返回一个索引向量，指示 x 的每个元素在 v 中的位置。

Answer 1

函数 match 适用于向量：

x <- sample(1:10)
x
# [1]  4  5  9  3  8  1  6 10  7  2
match(c(4,8),x)
# [1] 1 5

match 仅根据您的要求返回匹配的第一次遇到。它返回第一个参数中的值在第二个参数中的位置。

对于多重匹配，%in% 是正确的方法：

x <- sample(1:4,10,replace=TRUE)
x
# [1] 3 4 3 3 2 3 1 1 2 2
which(x %in% c(2,4))
# [1]  2  5  9 10

%in% 返回一个逻辑向量，只要第一个参数为 TRUE 如果可以在第二个参数中找到该值，否则返回 FALSE。

Answer 2

funprog {base} 中的函数 Position 也可以完成这项工作。它允许您传递任意函数，并返回第一个或最后一个匹配项。

位置(f, x, right = FALSE, nomatch = NA_integer)

Answer 3

关于上述方法效率的一个小注释：

 library(microbenchmark)

  microbenchmark(
    which("Feb" == month.abb)[[1]],
    which(month.abb %in% "Feb"))

  Unit: nanoseconds
   min     lq    mean median     uq  max neval
   891  979.0 1098.00   1031 1135.5 3693   100
   1052 1175.5 1339.74   1235 1390.0 7399  100

所以，最好的方法是

    which("Feb" == month.abb)[[1]]

Answer 4

是的，我们可以按如下方式查找向量中元素的索引：

> a <- c(3, 2, -7, -3, 5, 2)
> b <- (a==-7)  # this will output a TRUE/FALSE vector
> c <- which(a==-7) # this will give you numerical value
> a
[1]  3  2 -7 -3  5  2
> b
[1] FALSE FALSE  TRUE FALSE FALSE FALSE
> c
[1] 3

这是查找向量中元素索引的最有效方法之一。

Answer 5

library(vctrs)

vec_match(c("a", "y"), letters)
# [1]  1 25

vec_match() 的工作方式与基本 R match() 类似，但有一个明显的区别。它可以更好地控制 NA 匹配：

library(vctrs)

match(NA, c(NA, 1:2))
# [1] 1

vec_match(NA, c(NA, 1:2))
# [1] 1

vec_match(NA, c(NA, 1:2), na_equal = FALSE)
# [1] NA