R语言：计算累积和达指定值的元素个数及代码报错排查

需求：计算向量myvec = seq(0,1,0.1)的累积和达到特定值时，一共用了多少个元素。

初始编写的累积和函数 #

cumsum_for <- function(x)
{
  y = 1
  for(i in 2:length(x))  # 暂不考虑x长度为1或0的情况
    {x[i] = x[i-1] + x[i]
    y = y+1}
  return(y)
}

添加阈值参数后的修改版 #

为了实现“累积和达到阈值即停止计算”的逻辑，添加limit参数修改函数：

cumsum_for <- function(x, limit)
{
  y = 1
  for(i in 2:length(x))  # 暂不考虑x长度为1或0的情况
    {x[i] = x[i-1] + x[i]
    if(x >= limit) break
    y = y+1}
  return(y)
}

运行时的错误与警告 #

执行以下代码时出现不符合预期的结果和警告：

myvec = seq(0,1,0.1)
cumsum_for(myvec, 0.9)
[1] 10
Warning messages:
1: In if (x >= limit) break :
  the condition has length > 1 and only the first element will be used
[...]

问题分析 #

警告的核心原因是if(x >= limit)这一行的逻辑错误：这里的x是整个输入向量，长度大于1，但if语句仅能判断长度为1的逻辑值，因此R只会取向量的第一个元素进行判断，导致循环无法在累积和达到阈值时提前终止，最终返回了整个向量的元素数量10。

正确的逻辑应该是判断当前循环计算出的累积和（即x[i]）是否达到阈值，而非对整个向量进行判断。

修正后的函数 #

cumsum_for <- function(x, limit)
{
  # 处理边界情况：第一个元素就满足阈值条件
  if (x[1] >= limit) {
    return(1)
  }
  y = 1
  for(i in 2:length(x))  {
    x[i] = x[i-1] + x[i]
    y = y + 1
    # 判断当前计算出的累积和是否达到阈值，满足则终止循环
    if(x[i] >= limit) break
  }
  return(y)
}

测试验证 #

运行修正后的函数：

myvec = seq(0,1,0.1)
cumsum_for(myvec, 0.9)

返回结果为5，符合预期——因为前5个元素的累积和为0+0.1+0.2+0.3+0.4=1.0，首次达到并超过阈值0.9。

内容的提问来源于stack exchange，提问作者Saraha