防止某些元素在 for 循环中被 `sample()`ed

Question

下面我的for 循环使用time_to_remove 中的元素随机删除它们study。要删除的time 的可用元素是：2, 3, 4。

但问题不大。 time 元素有顺序。因此，在 sample() 删除之后，我们不能以：time = 0,1,3 或 time = 0,1,2,4 等结束。我们只能以：time = 0,1、time = 0,1,2、time=0,1,2,3 或 time=0,1,2,3,4 结束。

我想知道这样的条件是否可以添加到我下面的当前代码中？

library(tidyverse)
data <- expand_grid(study=1:4,group=1:2, time=0:4)

n = 1
time_to_remove <- unique(data$time)[-(1:2)]
unique_study <- unique(data$study)

for(i in time_to_remove) {
  studies_to_remove = sample(unique_study, size = n)
  time_to_remove = i
  unique_study <- setdiff(unique_study, studies_to_remove)
  data <- data %>%
    filter(
      !(   study %in% studies_to_remove &
            time %in% time_to_remove ))
}
data %>% as.data.frame()

Answer 1

我知道这不是答案。但是提问者想知道如果他尝试使用我的评论，代码会是什么样子：

for(i in time_to_remove) {
  studies_to_remove = sample(unique_study, size = n)
     # time_to_remove = i
  unique_study <- setdiff(unique_study, studies_to_remove)
  data <- data %>%
    filter(
      !(   study %in% studies_to_remove &
            time > i ))
}