【发布时间】:2019-06-22 15:39:59
【问题描述】:
我已经在 SO 上寻找这个问题的答案,但没有找到解决我的问题的方法。
我有一个包含几列的数据框,每一列至少有一个 NA。这些列的名称存储在字符向量vars_na 中。对于其中的每一个,我想创建一个虚拟变量,如果该观察值缺失,则取值为 0,否则为 1。
下面有一个可重现的玩具示例和我到现在为止使用的代码:
# creation of toy dataset
iris[1:5, 1] <- rep(NA, 5)
iris[1:10, 4] <- rep(NA, 10)
vars_na <- c("Sepal.Length", "Petal.Width")
for(var in vars_na){
iris <- iris %>%
mutate(dummy = ifelse(is.na(!!var), 0, 1)) %>%
rename_at(c("dummy"), list(~paste0("dummyna_", var)))
# 'rename_at' is just to differentiate between the several dummies created,
# and it works correctly
}
问题是新创建的假人导致向量全是1,所以他们没有正确考虑缺失值;确实:
head(iris)
Sepal.Length Sepal.Width Petal.Length Petal.Width Species dummyna_Sepal.Length dummyna_Petal.Width
1 NA 3.5 1.4 NA setosa 1 1
2 NA 3.0 1.4 NA setosa 1 1
3 NA 3.2 1.3 NA setosa 1 1
4 NA 3.1 1.5 NA setosa 1 1
5 NA 3.6 1.4 NA setosa 1 1
6 5.4 3.9 1.7 NA setosa 1 1
但我想获得
Sepal.Length Sepal.Width Petal.Length Petal.Width Species dummyna_Sepal.Length dummyna_Petal.Width
1 NA 3.5 1.4 NA setosa 0 0
2 NA 3.0 1.4 NA setosa 0 0
3 NA 3.2 1.3 NA setosa 0 0
4 NA 3.1 1.5 NA setosa 0 0
5 NA 3.6 1.4 NA setosa 0 0
6 5.4 3.9 1.7 NA setosa 1 0
代码很简单,我相信它应该可以工作。我做错了什么?提前致谢。
【问题讨论】: