R:用指标值替换预测值

Question

我有一个矩阵，其中包含 1900 万个 ID 和 Rating 的预测观测值。我想替换实际 Rating 中 ID 的预测 Rating。

这是我想要实现的目标的演示:

train <- data.frame(1:10,c(1,2,NA,4,5,NA,7,8,NA,10))
names(train) <- c("ID", "Rating")
pred <- data.frame(1:10, c(1,1,3,4,2,4,5,6,7,1))
names(pred) <- c("ID", "Rating")

我希望最终的预测是:

pred <- data.frame(1:10, c(1,2,3,4,5,4,7,8,7,10))
names(pred) <- c("ID", "Rating")

有什么建议吗？

尝试:

Answer 1

只需使用[]括号和is.na对数据进行子集化即可。

> train$Rating[is.na(train$Rating)] = pred$Rating[is.na(train$Rating)]
> train
   ID Rating
1   1      1
2   2      2
3   3      3
4   4      4
5   5      5
6   6      4
7   7      7
8   8      8
9   9      7
10 10     10

或者，换个方式:

> pred$Rating[!is.na(train$Rating)] = train$Rating[!is.na(train$Rating)]
> pred
   ID Rating
1   1      1
2   2      2
3   3      3
4   4      4
5   5      5
6   6      4
7   7      7
8   8      8
9   9      7
10 10     10