创建一个函数来识别缺失值

Question

我正在尝试构建一个函数作为 R 中更大函数的一部分。有些部分工作正常，但其他部分则不然。这是给我带来问题的代码片段。

这部分函数旨在识别数据框中的变量是否丢失，然后生成一个新变量来记录该特定情况是否丢失或存在。我希望新变量具有后缀 .zero（q1 变为 q1_zero，q2 变为 q2_zero 等）。我可以毫无问题地生成后缀。创建新变量会导致一些问题。任何见解将不胜感激。

function1 <- function (x, data) {
  # new variable name
  temp <- paste (x, .zero, sep="", collapse = NULL)
  temp
  
  # is variable missing
  # I don't know if I should use this method or ifelse()
  data$temp [is.na (data$x)]<- 0
  data$temp [!is.na (data$x)]<- 1
 return (data$temp)
  }

Answer 1

您遇到了一些问题

• .zero 未定义，您需要带引号的字符串 ".zero"
• 您不能使用 $列名称存储在字符串中。您需要使用data[[temp]]而不是data$temp。 如果您想了解更多信息，请参阅以下相关常见问题解答。
• 您可能想要返回整个修改后的数据框，而不仅仅是您添加的列（我假设这是因为您将整个数据框传递给了函数）。

我们还可以做一些简化，paste0()是paste(sep = "")和as.integer(!is.na(data$)的快捷方式x)) 是一种更干净、更有效的创造价值观的方式。

将所有这些放在一起：

function1 <- function (x, data) {
  data[[paste0(x, ".zero")]] = as.integer(!is.na(data[[x]]))
  return(data)
}

我想添加一点注释来说明 .zero 后缀对于值是否缺失而言并不是特别有用。更好的后缀可能类似于 .present —— 1 表示该值存在，0 表示不存在。

同样，对于函数来说，function1 绝对是一个糟糕的名称。使用描述性名称。 add_present_column 会是一个更好的名字。 （通常最好给函数命名为动词。）

由于我看到 Konrad 编辑了问题，我还会提到 R 函数中不需要 return() 。函数的最后一行将被返回，从风格上讲，许多人更喜欢函数的最后一行只是 data 而不是 return(data)。