根据 R 中的阈值对数字列表中的元素进行分组

Question

假设我有一个数字列表：

numbers = c(1, 2, 5, 6, 3, 2, 3, 7, 6)

和一个阈值

threshold = 6

我想根据阈值对数字列表中的元素进行分组——对元素求和，直到达到/超过阈值。保存每个组的第一个元素的索引，所以想要的输出是：

1, 4, 5, 8, 9

1 因为它是开始

4 因为 1+2+5 = 8 > 6，这包括 3 个元素

它类似于这篇文章Sum list of numbers until threshold，但我们希望继续添加，直到到达列表末尾。

Answer 1

您可以使用 MESS 包中的函数 cumsumbinning ：

numbers = c(1, 2, 5, 6, 3, 2, 3, 7, 6)
threshold = 6
which(!duplicated(MESS::cumsumbinning(numbers, threshold - 1, cutwhenpassed=TRUE), fromLast = TRUE))
#[1] 3 4 7 8 9

如果你想要组中的第一个元素。

which(!duplicated(MESS::cumsumbinning(numbers, threshold - 1, cutwhenpassed=TRUE)))
#[1] 1 4 5 8 9

---

cumsumbinning 和 cutwhenpassed=TRUE 会在值通过 threshold 值时创建一个新组，因此我使用了threshold - 1。

MESS::cumsumbinning(numbers, threshold - 1, cutwhenpassed=TRUE)
#[1] 1 1 1 2 3 3 3 4 5

Answer 2

我认为这也可以达到目的

library(purrr)
setdiff(which(accumulate(numbers, ~ifelse(.x + .y <  6, .y + .x, 0)) ==0), which(numbers == 0))

#check on new vector numbers
numbers = c(1, 2, 5, 6, 3, 2, 3, 7, 6, 0, 2, 3, 7, 1, 2)

setdiff(which(accumulate(numbers, ~ifelse(.x + .y <  6, .y + .x, 0)) ==0), which(numbers == 0))
[1]  3  4  7  8  9 13