实现不使用length()的向量长度计算函数,兼容numeric(0)输入
修复自定义向量长度计算函数
你的问题出在:当输入numeric(0)时,seq_along(x)会返回空的整数向量integer(0),而max()处理空向量会返回-Inf,不是预期的0。
下面是两种可行的修复方案:
方案1:给max添加兜底的0
通过把0和seq_along(x)合并成一个向量再取最大值。当seq_along(x)为空时,最大值就是0;非空时则取向量的真实长度:
my_length <- function(x){ max(c(0, seq_along(x))) }
方案2:条件判断空向量
直接判断输入是否为空向量,为空时返回0,否则按原逻辑执行:
my_length <- function(x){ if (length(x) == 0) { 0 } else { max(seq_along(x)) } }
测试两种方案的效果:
my_length(numeric(0)) # 返回0 my_length(c(1,3,5)) # 返回3 my_length(letters[1:4])# 返回4
内容的提问来源于stack exchange,提问作者Chuck Davidson
相关产品推荐
相关产品推荐

