Select根据R dplyr中的多列，每组中具有最大值的行

Question

我的数据框看起来像这样

library(tidyverse)

df1 <- tibble(col1= c("apple","apple","banana","banana"), 
              col2 = c("appl","aple","banan","bananb"),
              count_col1=c(1,1,4,4), count_col2=c(3,4,1,1))
df1
#> # A tibble: 4 × 4
#>   col1   col2   count_col1 count_col2
#>   <chr>  <chr>       <dbl>      <dbl>
#> 1 apple  appl            1          3
#> 2 apple  aple            1          4
#> 3 banana banan           4          1
#> 4 banana bananb          4          1

^{由 reprex package (v2.0.1)}

创建于 2022-02-17

我想 select 在 grouping_by col1 之后基于 count_col1 和 count_col2.

具有最大值的行

我希望我的数据看起来像这样

      col1   col2   count_col1 count_col2
      apple  aple            1          4
      banana banan           4          1
      banana bananb          4          1

一个专栏你可以写点东西

df1 %>% 
  slice(which.max(count_col1))

但不是两个人

Answer 1

我们可能会得到 'count' 列的按行最大值 pmax，按 'col1' 分组，filter max 值为 'Max' 列是。

library(dplyr)
df1 %>% 
 mutate(Max = pmax(count_col1, count_col2) ) %>%
 group_by(col1) %>%
 filter(Max == max(Max)) %>%
 ungroup %>%
 select(-Max)

-输出

# A tibble: 3 × 4
  col1   col2   count_col1 count_col2
  <chr>  <chr>       <dbl>      <dbl>
1 apple  aple            1          4
2 banana banan           4          1
3 banana bananb          4          1

我们也可以用slice_max

library(purrr)
df1 %>%
  group_by(col1) %>%
  slice_max(invoke(pmax, across(starts_with("count")))) %>%
  ungroup
# A tibble: 3 × 4
  col1   col2   count_col1 count_col2
  <chr>  <chr>       <dbl>      <dbl>
1 apple  aple            1          4
2 banana banan           4          1
3 banana bananb          4          1

Select根据R dplyr中的多列，每组中具有最大值的行

Select the row with the maximum value in each group based on multiple columns in R dplyr

r

dplyr

tidyverse