对向量元素求和，直到遇到正元素 - R答案

【问题标题】：Summing Vector elements until a Positive element is encountered - R对向量元素求和，直到遇到正元素 - R
【发布时间】：2014-01-31 18:33:44
【问题描述】：

我是 R 新手，我正在浏览一本名为“使用 R 发现统计数据”的书。虽然这本书暗示你不需要任何统计背景，但有些内容没有涵盖/解释......

我试图从位置 1 开始对向量的元素求和，直到出现正元素。

我发现this 问题与我想要实现的目标非常相似。但是，当我实现它时，它似乎并不总是有效（有时它似乎包含第一个积极元素）......

我的程序是：

  vecA <- runif(10, -10, 10);
  sumA <-sum(vecA [1:min(which(vecA < 0))]);

有没有更稳健的方法来计算这个而不使用每次都有效且不添加正元素的循环？我还没有进入我的书循环阶段。

我还发现this 网站提出了类似的问题，但他们的回答错误：

  sum(vecA [seq_len(which.max(vecA > 0)]);

【问题讨论】：

标签： r sum

【解决方案1】：

您可以使用以下代码：

sum(vecA * !cumsum(vecA > 0))

如果第一个元素是正数或所有元素都是负数，这也适用。

【讨论】：

【解决方案2】：

您想使用 > 而不是

您目前从 1 求和，直到达到第一个负值（包括第一个负值）。

sum(vecA[1:min(which(vecA>0))-1])

which() 函数将返回正元素的所有位置，然后将总和从 1 到第一个正元素的位置 - 1 将保证您对所有负元素求和

【讨论】：

【解决方案3】：

match 函数通常是在向量中找到第一次出现的某个元素的最快方法，因此它的另一个版本可能如下所示：

first.positive <- match(TRUE, vecA > 0)
sumA <- sum( vecA[ 1 : first.positive ] ) - vecA[first.positive]

如果正元素是第一个，这将给你零。

【讨论】：