用整数矩阵对 data.frame 进行子集化

我一直遇到这个问题，想知道是否有一个简单的解决方法。对于某些情况，我发现考虑将矩阵子集化更合乎逻辑

N <- 12
N.NA <- 6
dat <- data.frame(V1=runif(N),V2=runif(N))
sel.mat <- matrix(c(sample(seq(N),N.NA),sample(ncol(dat),N.NA,replace=TRUE)),ncol=2)

这适用于选择，但不适用于替换：

> dat[sel.mat]
[1] 0.2582569 0.8455966 0.8828083 0.5384263 0.9574810 0.5623158
> dat[sel.mat] <- NA
Error in `[<-.data.frame`(`*tmp*`, sel.mat, value = NA) : 
  only logical matrix subscripts are allowed in replacement

我意识到错误消息是有原因的（如果有多个替换指向同一元素，它不知道该怎么做），但这并不能阻止 R 允许对向量进行整数替换（例如dat$V1[c(2,3)] <- NA).

有没有一种方便的方法可以用整数矩阵替换？

将其转换为矩阵：

dat.m <- as.matrix(dat)
dat.m[sel.mat] <- NA
> dat.m
             V1         V2
 [1,] 0.2539189         NA
 [2,] 0.5216975         NA
 [3,] 0.1206138 0.14714848
 [4,] 0.2841779 0.52352209
 [5,] 0.3965337         NA
 [6,] 0.1871074 0.23747235
 [7,] 0.2991774         NA
 [8,]        NA 0.09509202
 [9,] 0.4636460 0.59384430
[10,] 0.5493738 0.92334630
[11,] 0.7160894         NA
[12,] 0.9568567 0.80398264

Edit解释一下为什么我们的 data.frame 出现错误

dat.m[sel.mat] <- NA

相当于执行以下操作：

temp <- dat
dat <- "[<-"(temp, sel.mat, value=NA)

 Error in `[<-.data.frame`(temp, sel.mat, value = NA) : 
 only logical matrix subscripts are allowed in replacement

现在我可以执行以下操作并且它有效：

dat <- "[<-"(as.matrix(temp), sel.mat, value=NA)