r - 从第一次出现 Pattern 到结束设置值 NA

Question

是否有更快/更短的方法来设置值，包括匹配到 NA 之后？

vec <- 1:10;vec[c(3,5,7)]<-c(NA,NaN,"remove")
#"1"      "2"      NA       "4"      "NaN"    "6"      "remove" "8"      "9"      "10"    

期望的结果:

#"1"      "2"      NA       "4"      "NaN"    "6"      NA       NA       NA       NA

我的代码:

vec[{grep("^remove$",vec)[1]}:length(vec)]<-NA

请注意:

在那种情况下，我们假设会有一个突出的“删除”元素。因此解决方案不必处理没有任何问题的情况。

Answer 1

您可以使用 match 在找到第一个匹配项后停止搜索:

m = match("remove", vec) - 1L
if (is.na(m)){
  vec 
} else {
  c(head(vec, m), rep(vec[NA_integer_], length(vec)-m))
}

不过，我猜你必须有一个相当大的矢量才能注意到速度差异。或者，这可能会更快:

m = match("remove", vec)
if (!is.na(m)){
  vec[m:length(vec)] <- NA 
}