将NA与model.matrix一起使用

Question

假设我有一个变量，我想从中生成虚拟变量。 model.matrix非常有帮助

model.matrix(~iris$Species-1)

但是说我缺少价值

iris$Species[1]=NA
model.matrix(~iris$Species-1)

这两个model.matrix输出之间的唯一区别是，第二个输出省略了带有NA的行。 我希望model.matrix将NA视为唯一值，并为NA创建一个附加列。

有没有一种有效的方法可以做到这一点，或者我必须编写一个函数来做到这一点？

Answer 1

一种解决方案可能是将您感兴趣的变量转换为一个因子，并且这样做时不要排除NA：

iris$Species[1] <- NA
mm2 <- model.matrix(~factor(iris$Species, exclude=NULL)-1)
>dim(mm2)
150   4