如何用dplyr计算前N行累积均值并向下填充整列?
使用dplyr实现前3个观测值均值填充整列
需求说明
给数据框新增一列,先计算目标列前3个观测值的均值,再把这个均值填充到新列的所有行(包括第4行及之后的行)。
修正后的实现代码
你原来的写法有个小问题:cummean(head(var1, 3))生成的是长度为3的累积均值向量,虽然R会自动循环填充至整列,但更直接准确的方式是直接计算前3个值的均值,再赋值给整列。
library(tidyverse) set.seed(1) dat <- data.frame(var1 = round(rnorm(10, 100, 20))) dat <- dat %>% mutate(var1_mean = mean(head(var1, 3)))
代码解析
head(var1, 3):提取var1列的前3个观测值mean():计算这3个值的算术平均值mutate():新增var1_mean列,将计算得到的均值赋值给该列的每一行
结果验证
运行代码后,dat$var1_mean的所有行都会是前3个值的均值(示例里前3个值是94、88、107,均值为96.333...),完全符合需求。
内容的提问来源于stack exchange,提问作者user13973103
相关产品推荐
相关产品推荐

