首页 > 解决方案 > 带示例的条件 group_by

问题描述

我的任务是识别数据集中的唯一试验 (1,2,3,...)。这是一个例子:

"source","ID","cultivar","design"
"PDMR_vol_12","CF027","Ambassador","RCBD"
"PDMR_vol_12","CF027","Ambassador","RCBD"
"PDMR_vol_12","CF027","Ambassador","RCBD"
"PDMR_vol_12","CF027","Ambassador","RCBD"
"PDMR_vol_7","CF026","ASG2000","RCBD"
"PDMR_vol_7","CF026","ASG2000","RCBD"
"PDMR_vol_7","CF026","ASG2000","RCBD"
"PDMR_vol_7","CF026","P26R61","RCBD"
"PDMR_vol_7","CF026","P26R61","RCBD"
"PDMR_vol_7","CF026","P26R61","RCBD"
"PDMR_vol_4","CF011","Roane","SP"
"PDMR_vol_4","CF011","Roane","SP"
"PDMR_vol_4","CF011","Tomahawk","SP"
"PDMR_vol_4","CF011","Tomahawk","SP"
"PDMR_vol_4","CF011","Everest","SP"
"PDMR_vol_4","CF011","Everest","SP"

条件列是:

unique_trials_RCBD<- ("source","ID","cultivar","design")

unique_trials_SP<-unique_trials_RCBD[-3]

使用基于几列的条件 group_by,我们几乎得到了正确的结果,除了它没有正确地将 (PDMR_vol_7 CF026) 识别为两次试验。

doAGroupBy <- function(data, some_condition) {

 if (some_condition == TRUE) {

   group_args <- unique_trials_RCBD

  } else {

   group_args <- unique_trials_SP

 }

  data %>%
    group_by_at(vars(group_args))
}


 a<-doAGroupBy(data, FALSE) %>% 
   mutate(trial_number=cur_group_id())

总共应该有 4 次试验。关于如何改进此代码的任何想法?谢谢

标签: rif-statementdplyrgroup-bytidyverse

解决方案


如果我正确理解了这个问题,这应该可以工作:

数据

df <-
tibble::tribble(~`source`, ~`ID`,~`cultivar`,~`design`,
  "PDMR_vol_12", "CF027", "Ambassador",  "RCBD",
  "PDMR_vol_12", "CF027", "Ambassador",  "RCBD",
  "PDMR_vol_12", "CF027", "Ambassador",  "RCBD",
  "PDMR_vol_12", "CF027", "Ambassador",  "RCBD",
   "PDMR_vol_7", "CF026",    "ASG2000",  "RCBD",
   "PDMR_vol_7", "CF026",    "ASG2000",  "RCBD",
   "PDMR_vol_7", "CF026",    "ASG2000",  "RCBD",
   "PDMR_vol_7", "CF026",     "P26R61",  "RCBD",
   "PDMR_vol_7", "CF026",     "P26R61",  "RCBD",
   "PDMR_vol_7", "CF026",     "P26R61",  "RCBD",
   "PDMR_vol_4", "CF011",      "Roane",    "SP",
   "PDMR_vol_4", "CF011",      "Roane",    "SP",
   "PDMR_vol_4", "CF011",   "Tomahawk",    "SP",
   "PDMR_vol_4", "CF011",   "Tomahawk",    "SP",
   "PDMR_vol_4", "CF011",    "Everest",    "SP",
   "PDMR_vol_4", "CF011",    "Everest",    "SP"
  ) 

代码

df %>% 
  # Creating auxiliar variable, consdering cultivar only for a RCBD design
  mutate(aux = if_else(design == "RCBD", cultivar,NA_character_)) %>%
  # Groupinp by source,ID,design and aux
  group_by(source,ID,design,aux) %>% 
  # Creating index grouped by variables above
  mutate(trial = group_indices())

结果

# A tibble: 16 x 6
# Groups:   source, ID, design, aux [4]
   source      ID    cultivar   design aux        trial
   <chr>       <chr> <chr>      <chr>  <chr>      <int>
 1 PDMR_vol_12 CF027 Ambassador RCBD   Ambassador     1
 2 PDMR_vol_12 CF027 Ambassador RCBD   Ambassador     1
 3 PDMR_vol_12 CF027 Ambassador RCBD   Ambassador     1
 4 PDMR_vol_12 CF027 Ambassador RCBD   Ambassador     1
 5 PDMR_vol_7  CF026 ASG2000    RCBD   ASG2000        3
 6 PDMR_vol_7  CF026 ASG2000    RCBD   ASG2000        3
 7 PDMR_vol_7  CF026 ASG2000    RCBD   ASG2000        3
 8 PDMR_vol_7  CF026 P26R61     RCBD   P26R61         4
 9 PDMR_vol_7  CF026 P26R61     RCBD   P26R61         4
10 PDMR_vol_7  CF026 P26R61     RCBD   P26R61         4
11 PDMR_vol_4  CF011 Roane      SP     NA             2
12 PDMR_vol_4  CF011 Roane      SP     NA             2
13 PDMR_vol_4  CF011 Tomahawk   SP     NA             2
14 PDMR_vol_4  CF011 Tomahawk   SP     NA             2
15 PDMR_vol_4  CF011 Everest    SP     NA             2
16 PDMR_vol_4  CF011 Everest    SP     NA             2

推荐阅读