將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