找到最后4种元素平均(Find average of 4 last elements)

我的数据集的形式如下：

df<- data.frame(c("a", "a", "a", "a", "a", "a", "a", "a", "b", "b", "b", "b", "b", "b", "b", "b"),
                c(1,    1,   1,   1,   2,   2,   2,   2,   1,   1,    1,  1,   2,    2,   2,   2),
                c(1,    2,   3,   4,   1,   2,   3,   4,   1,   2,   3 , 4,  1,    2,   3,   4),
                c(25,   75,  20,  40,  60,  50,  20,  10,  20,  30,  40,  60, 25,   75,  20,  40))
colnames(df)<-c("car", "year", "mnth", "val")

为了清楚我在这里展示它还有：

   car year mnth val
1    a    1    1  25
2    a    1    2  75
3    a    1    3  20
4    a    1    4  40
5    a    2    1  60
6    a    2    2  50
7    a    2    3  20
8    a    2    4  10
9    b    1    1  20
10   b    1    2  30
11   b    1    3  40
12   b    1    4  60
13   b    2    1  25
14   b    2    2  75
15   b    2    3  20
16   b    2    4  40

我想补充一个新列tmp到df就某一特定行，值tmp应该是平均df$val和3个前述值。 的一些例子tmp在这里显示

#row 3: mean(25,75,20)=40
#row 4: mean(25,75,20,40)=40
#row 5: mean(75,20,40,60)=48.75
#row 16: mean(25,75,20,40)=40

是否有一个有效的方式来做到这一点R中不使用for -loops？

Answer 1:

对于每一个值，计算滚动窗口，其包括值以及前述的3个值（从索引的平均值i-3到索引i在下面的溶液中）。 对于情况下，当i-3是否定的，你可以用0 （ max((i-3),0)

sapply(seq_along(df$val), function(i)
      mean(df$val[max((i-3),0):i], na.rm = TRUE))
#[1] 25.00 50.00 40.00 40.00 48.75 42.50 42.50 35.00 25.00
#[10] 20.00 25.00 37.50 38.75 50.00 45.00 40.00

还要考虑rollmean的zoo

library(zoo)
c(rep(NA,3), rollmean(x = df$val, k = 4))
#[1]    NA    NA    NA 40.00 48.75 42.50 42.50 35.00 25.00 20.00 25.00
#[12] 37.50 38.75 50.00 45.00 40.00
#FURTHER TWEAKING MAY BE NECESSARY

Answer 2:

这里的（有点）使用量化的解决方案data.table::shift

library(data.table)
colMeans(do.call(rbind, shift(df$val, 0:3)), na.rm = TRUE)
## [1] 25.00 50.00 40.00 40.00 48.75 42.50 42.50 35.00 25.00 20.00 25.00 37.50 38.75 50.00 45.00 40.00

---

或者，正如@Frank建议

rowMeans(setDF(shift(df$val, 0:3)), na.rm = TRUE)

Answer 3:

或者干脆这样

library(dplyr)
df$tmp <- (df$val+lag(df$val,1)+lag(df$val,2)+lag(df$val,3))/4

这不使用任何循环。 它只是转移列表，总结移位名单。

例如，如果你定义

a <- c(1,2,3,4,5)

然后

lag(a) 

是

NA  1  2  3  4

我希望它可以帮助你。

Answer 4:

你也可以使用data.table

library(data.table)

setDT(df)
df[, tmp := (val + shift(val,1,type="lag") + shift(val,2,type="lag") + shift(val,3,type="lag"))/4]

文章来源: Find average of 4 last elements