cur_group_id by size rather than alphabetical order

user_8453123 0
Viewed 260

I have the following data frame:

df <- structure(list(s_do_h_patients_state = c("NC", "NC", NA, NA, 
"MN", "MN", "UT", "UT", "IL", "IL"), diabetes = c(FALSE, TRUE, 
FALSE, TRUE, FALSE, TRUE, FALSE, TRUE, FALSE, TRUE), n = c(24191L, 
5684L, 24386L, 3820L, 18768L, 2423L, 19732L, 1313L, 15670L, 2336L
), p = c(0.809740585774059, 0.190259414225941, 0.864567822449124, 
0.135432177550876, 0.88565900618187, 0.11434099381813, 0.937609883582799, 
0.0623901164172012, 0.870265467066533, 0.129734532933467), N = c(29875L, 
29875L, 28206L, 28206L, 21191L, 21191L, 21045L, 21045L, 18006L, 
18006L)), row.names = c(NA, -10L), class = c("tbl_df", "tbl", 
"data.frame"))

I would like to add another column enumerating the groups, such that the output would be c(1,1,2,2,3,3...).

One way to do it would be group_indices, but this orders alphabetically rather than by group size. What's the correct way to achieve this?

dplyr grouping r
2 Answers

Here is a straightforward solution to this problem

library(dplyr)
df %>% mutate(group = match(s_do_h_patients_state, unique(s_do_h_patients_state)))

Output

# A tibble: 10 x 6
   s_do_h_patients_state diabetes     n      p     N group
   <chr>                 <lgl>    <int>  <dbl> <int> <int>
 1 NC                    FALSE    24191 0.810  29875     1
 2 NC                    TRUE      5684 0.190  29875     1
 3 NA                    FALSE    24386 0.865  28206     2
 4 NA                    TRUE      3820 0.135  28206     2
 5 MN                    FALSE    18768 0.886  21191     3
 6 MN                    TRUE      2423 0.114  21191     3
 7 UT                    FALSE    19732 0.938  21045     4
 8 UT                    TRUE      1313 0.0624 21045     4
 9 IL                    FALSE    15670 0.870  18006     5
10 IL                    TRUE      2336 0.130  18006     5

Note that you can't use rleid because

> data.table::rleid(c("NC", "NC", "IL", "NC"))
[1] 1 1 2 3
user_108024990 
df %>% arrange(desc(N)) %>%
mutate(id = dense_rank(desc(N)))
user_123476000
Related