向量搜索算法

Question

我有以下问题。假设我有一个向量：

v = [1,2,3,4,5,1,2,3,4,...]

我想从矢量中顺序采样点，这些点的绝对磁速差高于之前采样点的阈值。假设我的阈值是2。

我从索引1开始，对第一个点1进行采样，然后在v3满足我的条件，然后对3进行采样(从3-1 >= 2开始)。然后3，新的采样点成为我检查的参考。下一个采样点是5，即v5。然后下一个点是1，它是v，6。

不幸的是，我在R中编写的代码耗时太长。基本上，我会重复扫描数组并查找匹配项。但我认为这种方法太天真了。我有一种感觉，我可以在数组中完成这项任务。不过我不知道怎么做到的。感谢您的帮助。我想我遇到的问题是下一个采样点的位置可以在数组中的任何位置，我需要从当前点扫描数组到末尾才能找到它。

谢谢。

Answer 1

我看不到没有循环的方法可以做到这一点，所以这里有一个：

my.sample <- function(x, thresh) {
   out <- x
   i <- 1
   for (j in seq_along(x)[-1]) {
      if (abs(x[i]-x[j]) >= thresh) {
         i <- j
      } else {
         out[j] <- NA
      }
   }
   out[!is.na(out)]
}

my.sample(x = c(1:5,1:4), thresh = 2)
# [1] 1 3 5 1 3

Answer 2

您可以使用一些递归在没有循环的情况下完成此操作：

vsearch = function(v, x, fun=NULL) {
  # v: input vector
  # x: threshold level

  if (!length(v) > 0) return(NULL)

  y = v-rep(v[1], times=length(v))
  if (!is.null(fun)) y = fun(y)

  i = which(y >= x)

  if (!length(i) > 0) return(NULL)
  i = i[1]

  return(c(v[i], vsearch(v[-(1:(i-1))], x, fun=fun)))
}

使用上面的向量：

> vsearch(c(1,2,3,4,5,1,2,3,4), 2, abs)
[1] 3 5 1 3