r - 1 在R中分别对训练和测试数据进行热编码

Question

我需要根据前一个的长度添加大约100额外的列data.framedata.frame

例如，我有两个data.framesXtrain和Xtest. Xtrain作为 1000 列，但Xtest只有 900 列。这种差异是由于 1-hot 分别对Xtrainand进行编码Xtest。

如何将这 100 个缺失的列（全为 0）添加到Xtest？此外，扩充中的列顺序Xtest应与 相同Xtrain。

这是我到目前为止所尝试的：

extra = setdiff(names(Xtrain), names(Xtest))

for (e in extra){
    Xtest$e <- 0
}

但这会添加一列e而不是extra.