在 dplyr 链中获取组内的分位数

Question

我想做什么：

library(tidyverse)
diamonds |> 
  group_by(cut) |> 
  mutate(qt_25 = some_ideal_func_25_pctile(price))

我想改变一个新列，为每个组获取价格的 25 个百分点。

例如对于剪切 'Ideal':

diamonds |> filter(cut == 'Ideal') |> pull(price) |> quantile()
     0%     25%     50%     75%    100% 
  326.0   878.0  1810.0  4678.5 18806.0

然后我希望在理想剪辑组的所有行中重复 878.0。

如何在我的第一个代码块的 dplyr 链中执行此操作？

Answer 1

您可以简单地使用 quantile()s probs 参数（感谢 Axeman）：

library(tidyverse)

diamonds %>%  
  group_by(cut) %>% 
  mutate(qt_25 = quantile(price, 0.25))

这个returns

# A tibble: 53,940 x 11
# Groups:   cut [5]
   carat cut       color clarity depth table price     x     y     z qt_25
   <dbl> <ord>     <ord> <ord>   <dbl> <dbl> <int> <dbl> <dbl> <dbl> <dbl>
 1  0.23 Ideal     E     SI2      61.5    55   326  3.95  3.98  2.43  878 
 2  0.21 Premium   E     SI1      59.8    61   326  3.89  3.84  2.31 1046 
 3  0.23 Good      E     VS1      56.9    65   327  4.05  4.07  2.31 1145 
 4  0.29 Premium   I     VS2      62.4    58   334  4.2   4.23  2.63 1046 
 5  0.31 Good      J     SI2      63.3    58   335  4.34  4.35  2.75 1145 
 6  0.24 Very Good J     VVS2     62.8    57   336  3.94  3.96  2.48  912 
 7  0.24 Very Good I     VVS1     62.3    57   336  3.95  3.98  2.47  912 
 8  0.26 Very Good H     SI1      61.9    55   337  4.07  4.11  2.53  912 
 9  0.22 Fair      E     VS2      65.1    61   337  3.87  3.78  2.49 2050.
10  0.23 Very Good H     VS1      59.4    61   338  4     4.05  2.39  912 
# ... with 53,930 more rows

在 dplyr 链中获取组内的分位数

Get quantiles within groups within a dplyr chain

r

dplyr