R语言ifelse语句返回数组长度与预期不符如何解决

问题原因 #

R 中的 ifelse 是向量化运算函数，其返回结果的长度严格等于判断条件的长度。你当前代码中的判断条件是 x > 5，x 是长度为1的标量，因此判断条件长度为1，ifelse 最终只会取第二个、第三个参数的第一个元素返回，所以即使你用rep生成了长度为6的向量，最终也仅能得到长度为1的"known"结果。

解决方案 #

• 如果你仅需要做单次标量判断，直接使用原生的if...else分支结构即可，示例代码如下：

x <- 6
if (x > 5) {
  rep("known", x)
} else {
  rep("unknown", x)
}

运行上述代码即可得到长度为6的预期输出。

• 如果你的使用场景需要处理向量化的判断条件（即判断条件是长度大于1的向量），可以将返回结果包装为列表后再展开，示例代码如下：

# 示例：同时判断多个x的取值
x_list <- c(6, 2, 4, 7)
res <- unlist(ifelse(x_list > 5, list(rep("known", x_list)), list(rep("unknown", x_list))))

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