我正在寻找一个函数，它接受一个向量并不断删除第一个值，直到向量的总和小于 20。返回剩余的值。
我尝试了 for 循环和 while 循环，但找不到解决方案。
vec <- c(3,5,3,4,3,9,1,8,2,5)
short <- function(vec){
for (i in 1:length(vec)){
while (!is.na((sum(vec)) < 20)){
vec <- vec[i+1:length(vec)]
#vec.remove(i)
}
}
预期的输出应该是：
1,8,2,5
小于 20。
【问题讨论】：
标签： r loops for-loop vector while-loop
我会选择Reduce
vec[Reduce(f = "+", x = vec, accumulate = T, right = T) < 20]
##[1] 1 8 2 5
或者，使用带有条件参数na.rm = T 的函数sum 定义Reduce，以便在需要时处理NA：
vec2 <- c(3, 2, NA, 4, 5, 1, 2, 3, 4, 9, NA, 1, 2)
vec2[Reduce(f = function(a,b) sum(a, b, na.rm = T), x = vec2, accumulate = TRUE, right = T) < 20]
##[1] 3 4 9 NA 1 2
我发现 Reduce 选项从右开始（整数向量的结尾），因此不必先反转它，方便。
【讨论】：
没有循环的基本解决方案
不是我最易读的代码，但它非常快（参见下面的基准测试）
rev( rev(vec)[cumsum( replace( rev(vec), is.na( rev(vec) ), 0 ) ) < 20] )
#[1] 1 8 2 5
注意：从@Ronak 的回答中“借用”NA-handling
样本数据vec = c(3, 2, NA, 4, 5, 1, 2, 3, 4, 9, NA, 1, 2)
基准测试
microbenchmark::microbenchmark(
Sotos = {
while (sum(vec, na.rm = TRUE) >= 20) {
vec <- vec[-1]
}
},
Ronak = tail(vec, sum(cumsum(replace(rev(vec), is.na(rev(vec)), 0)) < 20)),
Wimpel = rev( rev(vec)[cumsum( replace( rev(vec), is.na( rev(vec) ), 0 ) ) < 20]),
WimpelMarkus = vec[rev(cumsum(rev(replace(vec, is.na(vec), 0))) < 20)]
)
# Unit: microseconds
# expr min lq mean median uq max neval
# Sotos 2096.795 2127.373 2288.15768 2152.6795 2425.4740 3071.684 100
# Ronak 30.127 33.440 42.54770 37.2055 49.4080 101.827 100
# Wimpel 13.557 15.063 17.65734 16.1175 18.5285 38.261 100
# WimpelMarkus 7.532 8.737 12.60520 10.0925 15.9680 45.491 100
【讨论】：
rev：vec[rev(cumsum(rev(replace(vec, is.na(vec), 0))) < 20)]。这可能会进一步加快速度。
查看预期输出，您似乎想要删除值，直到剩余值的总和小于 20。
我们可以创建一个函数
drop_20 <- function(vec) {
tail(vec, sum(cumsum(rev(vec)) < 20))
}
drop_20(vec)
#[1] 1 8 2 5
在另一个输入上尝试
drop_20(1:10)
#[1] 9 10
分解功能，首先是vec
vec = c(3,5,3,4,3,9,1,8,2,5)
然后我们reverse 它
rev(vec)
#[1] 5 2 8 1 9 3 4 3 5 3
对其进行累积总和 (cumsum)
cumsum(vec)
#[1] 3 8 11 15 18 27 28 36 38 43
找出小于 20 的条目数
cumsum(rev(vec)) < 20
#[1] TRUE TRUE TRUE TRUE FALSE FALSE FALSE FALSE FALSE FALSE
sum(cumsum(rev(vec)) < 20)
#[1] 4
最后使用tail 对这些最后的条目进行子集化。
对代码稍作修改，它应该也能处理NAs
drop_20 <- function(vec) {
tail(vec, sum(cumsum(replace(rev(vec), is.na(rev(vec)), 0)) < 20))
}
vec = c(3, 2, NA, 4, 5, 1, 2, 3, 4, 9, NA, 1, 2)
drop_20(vec)
#[1] 3 4 9 NA 1 2
逻辑是我们 replace NA 用零然后取 cumsum
【讨论】：
NAs 你现在可以看看更新的答案。
您每次都需要删除第一个值，所以您的while 循环应该是，
while (sum(x, na.rm = TRUE) >= 20) {
x <- x[-1]
}
#[1] 1 8 2 5
【讨论】：
sum(x, na.rm = TRUE)
x <- x[-1]在R中的行为，所以可能更糟）可能是原因。
while 循环与不需要while。即使是这种情况，当我遵循 OP 的思路以显示错误时，投票仍然是有偏见的。这是一个糟糕的投票，但是人们会看到这一点。 x <- x[-1] 也只是删除第一个值...没有复杂的行为