r - IF 语句构造

Question

在 Excel 中，您可以创建一个 new_column，如果 old_column 为空，则将其定义为 0，如果 old_column 不为空，则将其定义为 1。

new_column=IF(ISBLANK([@[old_column]]),0,1)

任何人都可以想出一种有效的方法来对 r 中的数据框执行此操作。假设数据框中的一列称为old_column，我想使用上述描述向data_frame添加一个new_column。

我尝试过这个:

mydf$old_column[is.na(mydf$old_column] <- 0
mydf$old_column[!is.na(mydf$old_column] <- 1

但它给了我这个错误:

invalid factor level, NA generated error.

Answer 1

mydf$new_column <- as.integer(!is.na(mydf$old_column)

您甚至可能不需要 as.integer() 因为 is.na() 返回一个 bool 向量，而 R 处理 TRUE 并且FALSE 分别为 1 和 0。