后续：泛化一个data.frame子集函数2

Question

我正在跟进this great answer。在该答案中，foo2 函数帮助用户识别其中唯一的study，其他两个选定列（group&outcome）中的任何一个常量 em> 或 变化。

现在，假设我们想要另外确定是否存在唯一的study，其中所选变量之一 (outcome) 对于某些行是恒定的任何其他选定的变量 (group)。

例如，在study==14 中，outcome 对于group 的某些行是常量。但是在study==8 中，outcome 对于group 的所有行都是完全不同的：

   study group outcome
17    14     1       6
18    14     2       6
19    14     3       7
20    14     4       7

   study group outcome
9      8     1       2
10     8     2       3
11     8     3       4

有没有办法将foo2 扩展到另外识别像study==14 这样的研究？

dat = read.csv("https://raw.githubusercontent.com/rnorouzian/s/main/cf.csv")

study8 = subset(dat, study==8)[1:3]; study14 = subset(dat, study==14)[1:3] 

Answer 1

我们可以用case_when添加一些条件

foo2 <- function(dat, study_col, ...) {
   
    dot_cols <- ensyms(...)
    str_cols <- purrr::map_chr(dot_cols, rlang::as_string)

    dat %>%
           dplyr::select({{study_col}}, !!! dot_cols) %>%
            dplyr::group_by({{study_col}}) %>%
            dplyr::mutate(grp = across(all_of(str_cols), ~ {
                      tmp <- n_distinct(.)
                      case_when(tmp  == 1 ~ 1, tmp == n() ~ 2, tmp >1 & tmp < n() ~ 3,  TRUE ~ 4)
                      }) %>%
                   purrr::reduce(stringr::str_c, collapse="")) %>%
                dplyr::ungroup(.) %>%
                 dplyr::group_split(grp, .keep = FALSE)
           
     }

foo2(dat, study, group, outcome)