如何为Tibble添加id分组专属均值列?
实现分组均值列的几种方法
嘿,这事儿在R里有好几种简单靠谱的实现方式,我给你列几个最常用的:
1. 使用dplyr(适配tibble的tidyverse风格)
因为你用的是tibble,dplyr是tidyverse生态里的工具,用起来特别顺手:
library(dplyr) # 先定义你的原始数据 df <- tibble(id = c(1,1,2,2), data = c(1:4)) # 按id分组并添加均值列 df_with_means <- df %>% group_by(id) %>% # 按id分组 mutate(id_means = mean(data)) %>% # 给每个分组计算均值,填充到每一行 ungroup() # 可选:取消分组状态,避免后续操作受分组影响 print(df_with_means)
运行后就能得到你想要的结果,mutate会自动把每个分组的均值重复填充到该组的所有行里。
2. 使用base R的ave函数(无需额外安装包)
如果你不想加载额外的包,base R的ave()函数就能搞定:
df <- tibble(id = c(1,1,2,2), data = c(1:4)) df$id_means <- ave(df$data, df$id, FUN = mean) print(df)
ave()函数的作用就是对向量按指定分组计算函数值,返回和原向量长度一致的结果,完美匹配这个需求。
3. 使用data.table(适合大数据集)
如果你的数据量很大,data.table的效率会更高:
library(data.table) # 把tibble转换成data.table dt <- as.data.table(tibble(id = c(1,1,2,2), data = c(1:4))) # 按id分组添加均值列 dt[, id_means := mean(data), by = id] print(dt)
data.table的:=操作符可以直接在原表上添加列,不用复制数据,处理大数据时速度优势明显。
内容的提问来源于stack exchange,提问作者stollenm
相关产品推荐
相关产品推荐

