[英]Rolling sum in R
df <- data.frame(x = seq(1:10))
我要這個:
df$y <- c(1, 2, 3, 4, 5, 15, 20 , 25, 30, 35)
即每個y
是前五個x
值的總和。 這意味着前五個y
與x
相同
我得到的是:
df$y1 <- c(df$x[1:4], RcppRoll::roll_sum(df$x, 5))
x y y1
1 1 1
2 2 2
3 3 3
4 4 4
5 5 15
6 15 20
7 20 25
8 25 30
9 30 35
10 35 40
總而言之,我需要y
但我只能達到y1
1)增強的求和函數定義一個函數Sum
,如果它接收到6個值,則將其前5個值相加,否則返回最后一個值。 然后在rollapplyr
其與partial=TRUE
一起rollapplyr
:
Sum <- function(x) if (length(x) < 6) tail(x, 1) else sum(head(x, -1))
rollapplyr(x, 6, Sum, partial = TRUE)
## [1] 1 2 3 4 5 15 20 25 30 35
2)求和6並減去原始向量另一種可能性是取6個元素的連續求和並用NA填充前5個元素,然后減去原始向量。 最后填寫前5個。
replace(rollsumr(x, 6, fill = NA) - x, 1:5, head(x, 5))
## [1] 1 2 3 4 5 15 20 25 30 35
3)指定偏移量第三種可能性是使用width的偏移量形式來指定前5個元素:
c(head(x, 5), rollapplyr(x, list(-(1:5)), sum))
## [1] 1 2 3 4 5 15 20 25 30 35
4)偏移量的替代規范在此替代方案中,我們為前5個元素中的每一個元素指定0的偏移量,為其余元素指定-(1:5)的偏移量。
width <- replace(rep(list(-(1:5)), length(x)), 1:5, list(0))
rollapply(x, width, sum)
## [1] 1 2 3 4 5 15 20 25 30 35
填充前5個元素的方案似乎很不常見,您可能會考慮對前5個元素使用部分和,對第一個元素使用NA或0,因為沒有在先元素會生成該元素:
rollapplyr(x, list(-(1:5)), sum, partial = TRUE, fill = NA)
## [1] NA 1 3 6 10 15 20 25 30 35
rollapplyr(x, list(-(1:5)), sum, partial = TRUE, fill = 0)
## [1] 0 1 3 6 10 15 20 25 30 35
rollapplyr(x, 6, sum, partial = TRUE) - x
## [1] 0 1 3 6 10 15 20 25 30 35
一種簡單的方法是:
df <- data.frame(x = seq(1:10))
mysum <- function(x, k = 5) {
res <- rep(NA, length(x))
for (i in seq_along(x)) {
if (i <= k) { # edited ;-)
res[i] <- x[i]
} else {
res[i] <- sum(x[(i-k):(i-1)])
}
}
res
}
mysum(df$x)
# [1] 1 2 3 4 5 15 20 25 30 35
mysum <- function(x, k = 5) {
res <- x[1:k]
append<-sapply(2:(len(x)+1-k),function(i) sum(x[i:(i+k-1)]))
return(c(res,append))
}
mysum(df$x)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.