Rolling mean lag function for multiple variables

Question

Thanks to @josliber's help (Rolling mean lag function), I was able to speed up a rolling mean function for different groups and rollmean lengths.

Now I'd like to add to this function the ability to loop through different variables and bind everything together.

Minimum reproducible example

library(zoo)
dat <- data.frame(fips = rep(c(1001, 1003), each = 100),
 x = rnorm(200),
 x2 = rnorm(200),
 x3 = rnorm(200))
allFipsRM3 = function(dat, varName, len){
 do.call(rbind, lapply(split(dat, dat$fips), function(x) {
 all.rm <- as.data.frame(sapply(len, function(l) c(rollmean(x[,varName], l), rep(NA, l-1))))
 colnames(all.rm) <- paste0(varName, "_rm", len)
 cbind(data.frame(fips=x$fips[1]), all.rm, data.frame(year=seq_len(nrow(x))-1))
 }))
}
outdat3 <- allFipsRM3(dat, "x", c(1, 2))
 fips x_rm1 x_rm2 year
1001.1 1001 1.3482892 1.3043620 0
1001.2 1001 1.2604348 0.2990267 1
1001.3 1001 -0.6623813 -0.4243813 2
1001.4 1001 -0.1863812 0.2806624 3
1001.5 1001 0.7477061 -0.5111745 4
1001.6 1001 -1.7700551 -0.8463731 5

Answer 1

Answering my own question here, but certainly open to suggestions.

library(RcppRoll)
# Loop through n = 10
for (i in 1:10){
 # Create custom col labels
 lab1 <- paste0("x_", i)
 lab2 <- paste0("x2_", i)
 lab3 <- paste0("x3_", i)
 # Loop through each fips and calculate rollingmean
 dat <- dat %>%
 group_by(fips) %>%
 mutate(!!lab1 := roll_mean(x, i, align = "left", fill = "NA"),
 !!lab2 := roll_mean(x2, i, align = "left", fill = "NA"),
 !!lab3 := roll_mean(x3, i, align = "left", fill = "NA")) %>% 
 ungroup()
 # Progress bar for loop
 print(i)
}
> names(dat)
 [1] "fips" "x" "x2" "x3" "x_1" "x2_1" "x3_1" "x_2" "x2_2" "x3_2" "x_3" "x2_3" 
[13] "x3_3" "x_4" "x2_4" "x3_4" "x_5" "x2_5" "x3_5" "x_6" "x2_6" "x3_6" "x_7" "x2_7" 
[25] "x3_7" "x_8" "x2_8" "x3_8" "x_9" "x2_9" "x3_9" "x_10" "x2_10" "x3_10"