R：如果差异超过阈值，则累积和

Question

我有一个数字向量

x <- c(0, 28, 59, 89, 0, 15, 16, 0, 35, 31)
#[1] 0 0 31 30 0 15 16 0 35 31

对于特殊情况，我想计算它的累积总和。给定：

month_vec <- seq(as.Date("2009-02-01"), length = 10, by = "1 month") - 1
day_vec   <- as.numeric(substr(month_vec, 9, 10))
# > day_vec
#[1] 31 28 31 30 31 30 31 31 30 31

如果与之前元素的差异大于或等于day_vec 中的值，我只想cumsum(x)。

结果应该是这样的：

my_custom_cumsum(x)
#[1] 0 0 31 61 0 15 16 0 35 66

因为x[4]等于day_vec[4]，x[3]和x[4]是相加的。但是，x[6] 和 x[7] 不会累加，因为它们小于它们在 day_vec 中的相应位置。但是x[9] 和x[10] 应该再次被累加——换句话说：如果与之前元素的差值小于day_vec 中的值，则应该重置累加。有人知道如何优雅地解决这个问题吗？

Answer 1

我会使用用于子集的逻辑索引来执行此操作。对于应该“累积”的 x 的所有元素，它应该为真，其余元素为假。

idx <- x >= day_vec

现在您可以使用它来计算 cumsum 并将其分配给 x 中的正确元素：

x[idx] <- cumsum(x[idx])
x
#[1]  0  0 31 61  0 15 16