Search code examples
rdplyrgroupmutate

find the first TRUE within a group

I want to save the position of the first TRUE in level variable within each id. I am doing the below:

library(tidyverse)

data <- data %>%
  group_by(id) %>%
  mutate(first_true = min(which(data$level == TRUE))) %>%
  ungroup()

However, the result for first_true is the same for all id. group_by(id) is not working. Any idea of what might be happening?

Example data:

id <- c(1,1,1,2,2,2,3,3,3)
level <- c(FALSE,FALSE,FALSE,FALSE,TRUE,TRUE,TRUE,TRUE,TRUE)
data <- data.frame(id, level)
data$level <- as.logical(data$level)

I would like to obtain:

id level
1 0
1 0
1 0
2 2
2 2
2 2 
3 1
3 1
3 1
Solution

  • You can try

    data %>%
    mutate(first_true = ifelse(any(level), min(which(level)), 0), .by = id)

    which gives

      id level first_true
1  1 FALSE          0
2  1 FALSE          0
3  1 FALSE          0
4  2 FALSE          2
5  2  TRUE          2
6  2  TRUE          2
7  3  TRUE          1
8  3  TRUE          1
9  3  TRUE          1