替代 for 循环

Question

考虑以下示例数据：

mapmatrix <- cbind(c(1,2,3,4,5,6),c(1,2,4,5,7,10))
arrayA <- c(1,2,4,7,10)
arrayB = rep(0,length(arrayA))
for(i in 1:length(arrayB))
{
arrayB[i] = which(mapmatrix[,2]==arrayA[i])
}

我被告知要尽可能避免 R 中的 for 循环。即使arrayA 有大约 400K 个元素，并且mapmatrix 的尺寸为 {200 万 x 2}，我确实发现它非常耗时。如果我可以用更快的替代方案替换for-loop，将不胜感激。

Answer 1

你可以试试：

 match(arrayA, mapmatrix[,2])
 #[1] 1 2 3 5 6

如果你有重复匹配，fmatch from library(fastmatch) 会更快

  library(fastmatch)
  fmatch(arrayA, mapmatrix[,2])
  #[1] 1 2 3 5 6