r - 创建一个计数器，只计算某个逻辑值并给出相同的重复次数

Question

假设我有这样的逻辑值

rex <- c(TRUE, FALSE, TRUE, TRUE, FALSE, TRUE, NA)

如何创建一个计数器，该计数器仅对 TRUE 进行计数，并将连续重复分组为计数器中的相同值。前任。拥有这样的东西:

1, NA, 2,2,NA, 3, NA

Answer 1

接受的解决方案不适用于以下情况:

rex <- c(TRUE, FALSE, TRUE, TRUE, FALSE, TRUE, NA, NA, TRUE)

它打印1 NA 2 2 NA 3 NA NA NA。

以下函数返回正确答案:

solution <- function (rex) {
  result <- c()
  counter <- 1
  consecutive <- FALSE
  for (i in 1:length(rex)) {
    if (rex[i] == TRUE && !is.na(rex[i]) && consecutive) {
      result <- c(result, counter)
      consecutive <- TRUE
    } else if (rex[i] == TRUE && !is.na(rex[i]) && !consecutive) {
      result <- c(result, counter)
      consecutive <- TRUE
    } else{
      if(i < length(rex) && rex[i+1] == TRUE && !is.na(rex[i+1])){
        counter <- counter + 1
      }
      result <- c(result, NA)
      consecutive <- FALSE
    }
  }
  return(result)
}

调用 solution(rex) 打印 1 NA 2 2 NA 3 NA NA 4，这是正确的答案。