R中的加权采样

Question

我有一个数据帧data 。 在每一行i都分配了一个data$ww的权重。 现在我想制作一个示例new_data data ，加权df$ww 。 我尝试过subset但速度很慢。

# sample data
data <- data.frame(var1 = log(sample(1:5000)))
ndata <- nrow(data)
maxW <- max(data$var1)

nsample <- 4000
rr <- runif(ndata)
data$ww <- cumsum(exp(data$var1))
new_data <- data[0, ]
i <- 1
while(nrow(new_data) < nsample) {
  new_data[i, ] <- subset(data, data$ww > rr[i] * maxW)[1,]
  i <- i + 1
}

有更快的方法吗？

Answer 1

使用sample()的prob参数：

samp_idx <- sample(seq_len(nrow(data)), nsample, prob=data$ww)
new_data <- data[samp_idx, ]

像这样的东西。 运行时间是

# user  system elapsed 
# 0.015   0.000   0.014 

与您的版本：

# user  system elapsed 
# 4.278   0.007   4.290