CODEWITHSUNDEEP
rdplyrsubsetsummarize
eyei
eyei

Reputation: 402

How to subset inside summarise dplyr

I would like to subset inside summarise(). Is the following subset()-ing somehow possible?

df <- structure(list(category = structure(c(1L, 1L, 1L, 2L, 2L, 1L, 
                                        1L, 1L, 2L, 1L, 1L, 1L, 1L, 2L), .Label = c("category MB", "category LR"
                                        ), class = "factor"), start = c(111, 222, 333, 444, 555, 111, 
                                                                        222, 333, 444, 111, 111, 222, 333, 444), stop = c(666, 777, 888, 
                                                                                                                          999, 1000, 666, 777, 888, 999, 666, 666, 777, 888, 999), ID = c(101, 
                                                                                                                                                                                          101, 101, 101, 101, 102, 102, 102, 102, 102, 102, 102, 102, 102
                                                                                                                          )), row.names = c(NA, -14L), class = "data.frame")

library(dplyr)

df %>% 
group_by(ID) %>% 
summarise(
    countAll = n(),
    durationAll = sum(stop - start),
    countCategoryMB = sum(category == "category MB"),
    durationCategoryMB = sum( subset(., category == "category MB", select = stop) -  subset(., category == "category MB", select = start) ), # line in question, currently wrong
    countCategoryLR = sum(category == "category LR"),
    durationCategoryLR = sum( subset(., category == "category LR", select = stop) -  subset(., category == "category LR", select = start) ) # line in question, currently wrong
)

The expected result (pic at the end of the post), I am able to achieve with left_join(). But I hope that it is possible to achieve the desired output in one-call with something like the code above.

# expected result achieved with left_join()
 df %>%
  group_by(ID) %>%
  summarise(countAll = n(),
            durationALL = sum(stop - start)) %>%
  left_join(
            .,
            df %>%
            filter(category == "category MB") %>%
            group_by(ID) %>%
            summarise(
            countCategoryMB = n(),
            durationCategoryMB = sum(stop - start)
           ),
           by = "ID"
 ) %>%
 left_join(
           .,
           df %>%
           filter(category == "category LR") %>%
           group_by(ID) %>%
           summarise(
           countCategoryLR = n(),
           durationCategoryLR = sum(stop - start)
           ) ,
          by = "ID"
          )

enter image description here

Thank you for your time!

Upvotes: 1

Views: 2250

Answers (1)

Bas
Bas

Reputation: 4658

In the below solution, (category == "category MB") equals 1 if it is True, otherwise it is 0. Therefore this effectively only sums the values of start and stop for those rows where category equals "category MB" or "category LR", as requested.

df %>% 
  group_by(ID) %>% 
  summarise(
    countAll = n(),
    durationAll = sum(stop - start),
    countCategoryMB = sum(category == "category MB"),
    durationCategoryMB = sum( ((category == "category MB")*stop) - ((category == "category MB")*start) ),
    countCategoryLR = sum(category == "category LR"),
    durationCategoryLR = sum( ((category == "category LR")*stop) - ((category == "category LR")*start) )
)

Upvotes: 3

Related Questions