R语言使用列表项内dataframe的source_country首值重命名列表项

问题排查 #

你的代码存在3处逻辑错误：

• 初始判断逻辑反向：if (!is.null(dataset[[i]]))代表列表元素不为空时就进入打印分支，直接跳过了后续命名逻辑，和「为空时提示无数据」的需求完全相反
• 行号判断范围不全：第二个分支仅匹配行数大于1的数据集，你第二个测试数据只有1行，无法触发命名操作
• 缺少NA值转字符串的逻辑：原生NA值直接赋值给列表名不会自动转成你需要的字符串"NA"

修复后代码 #

循环版本（兼容你原有的写法） #

for (i in seq_along(dataset)) {
  # 空元素直接提示
  if (is.null(dataset[[i]])) {
    print("no data")
    next
  }
  # 只要有行就取source_country第一个值
  if (nrow(dataset[[i]]) >= 1) {
    first_val <- dataset[[i]]$source_country[1]
    names(dataset)[i] <- ifelse(is.na(first_val), "NA", as.character(first_val))
  }
}

简洁向量化版本（基于purrr包） #

library(purrr)
names(dataset) <- map_chr(dataset, ~{
  if (is.null(.x)) return("no_data")
  val <- .x$source_country[1]
  ifelse(is.na(val), "NA", as.character(val))
})

结果验证 #

运行上述代码后，执行names(dataset)即可得到预期输出：

[1] "D"  "NA"

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