在R中原始数据框中按组创建均值新列

如何解决在R中原始数据框中按组创建均值新列

我有一个如下数据框：

unit_id  outcome
1        3
1        5
1        4
2        1
2        2
2        3

我知道如何计算每个unit_id的平均值。

df <- df %>%
  group_by(unit_id) %>%
  summarise(mean = mean(outcome))

这将产生：

unit_id  mean 
1        4
2        2

我正在尝试找出一种方法来获取每个unit_id的均值并将其包括在原始数据帧中。我希望输出如下所示。

unit_id  outcome  mean
1        3        4
1        5        4
1        4        4
2        1        2
2        2        2
2        3        2

解决方法

我们可以使用mutate代替summarise

library(dplyr)
df <- df %>%
        group_by(unit_id) %>%
        mutate(mean = mean(outcome))