r - 达到累积总和的元素数

我需要输出(对于进一步的 zoo:roll_apply)要求和以达到某个值的元素数。下面是一个例子:

  # value to reach
  vTR = c(10,15,12,13,10,15,10)
  # element to sum
  element = c(9,6,5,2,1,9,1)
  magicFoo(vTR, element)
  # should return c(NA, 2, 3, 3, 4, 4, 2)
  # 10 ~ NA, 15 <= 9+6, 12 <= 5+6+9, 13 <= 2+5+6, 10 <= 1+2+5+6...

例如，我正在寻找某种 ave，其中 k 是动态计算的。

我可以用 for 循环来做到这一点，但我在这里寻找一种更优雅的方式来做到这一点:

vTR = c(10,15,12,13,10,15,10)
# element to sum
element = c(9,6,5,2,1,9,1)
res = c()
j = 1
k = 0
sumE = 0
for (i in 1:length(vTR)){
  k = k+1
  sumE = sum(element[j:k])
  if (sumE < vTR[i]) {
    res[length(res)+1] = NA
    next
  }
  repeat {
    j = j + 1
    sumE = sum(element[j:k])
    if (sumE < vTR[i]) {
      j = j-1
      res[length(res)+1] = k-j +1
      break
    }
  }
}

# > res
# [1] NA  2  3  3  4  4  2

最佳答案

使用 sapply 我们可以遍历 vTR 中的每个元素并获取第一个 x 值，将它们反转并对其进行累积总和，并在该值与 vTR[x] 值交叉时找到索引。

sapply(seq_along(vTR),function(x) which.max(cumsum(rev(head(element, x)))>=vTR[x]))

#[1] 1 2 3 3 4 4 2

为了获得确切的预期输出，我们可以通过

sapply(seq_along(vTR), function(x) {
  val = cumsum(rev(head(element, x)))
  if (sum(val)  >= vTR[x])
    which.max(val >= vTR[x])
  else
    NA
})

#[1] NA  2  3  3  4  4  2

关于r - 达到累积总和的元素数，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/49872475/