R语言编写函数判断dataframe的NA值并输出提示，现有代码报错如何修复

原代码报错核心原因 #

• 索引语法错误：order$count是一维向量，不能使用仅适用于二维数据的[,]索引语法，该写法会直接触发语法报错
• 判断逻辑错误：if条件必须传入单个布尔值，原代码直接将NA作为判断条件，无法完成逻辑判定
• 需求匹配错误：原需求为判断count列是否存在NA值，原代码未调用any()等聚合函数判断NA的存在性，逻辑完全不符合需求

修正后代码 #

整表校验版本（符合原需求：存在任意NA则输出Not valid，否则输出Success） #

order = data.frame(item = c("A","B","C"), count = c(1,2,NA))

check <- function(order){
  # 校验count列是否存在任意NA值
  if (any(is.na(order$count))) {
    print("Not valid")
  } else {
    print("Success")
  }
}

# 测试调用
check(order)

逐行校验版本（如需输出每个item的校验结果可使用） #

若需要针对每行item的count值单独返回校验结果，可使用向量化操作实现：

check_rowwise <- function(order){
  return(ifelse(is.na(order$count), paste0(order$item, ": Not valid"), paste0(order$item, ": Success")))
}

# 测试调用
check_rowwise(order)

运行说明 #

• 整表校验版本调用测试数据时，因count列存在NA值，输出结果为[1] "Not valid"；将测试数据中的NA替换为有效值后，输出结果为[1] "Success"
• 逐行校验版本调用测试数据时，会返回长度为3的字符向量，分别对应三个item的校验结果。

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