R中的滚动中位数答案

【问题标题】：Rolling median in RR中的滚动中位数
【发布时间】：2021-11-29 06:52:39
【问题描述】：

我需要计算滚动中位数。并有以下代码

i <- 0
median_roll<-c("")
x<-c(1:10)
n<-2
y<-as.data.frame(x)
  while(i < length(x)-n){
    median_roll[i] <- median(y[i:i+n,])
    i <- i + 1  
  }

在 median_roll 中产生以下 reluts

[1] "3" "4" "5" "6" "7" "8" "9"

我需要的是

[1] "2" "3" "4" "5" "6" "7" "8" "9"

【问题讨论】：

看起来 i:i+n 应该是 i:(i+n)。
@Miff 产生[1] "2" "3" "4" "5" "6" "7" "8"的输出
为什么median_roll<-c("")?!?中位数是一个数值，以median_roll<-c()开头。
单行：zoo::rollapplyr(y$x, width = 3L, FUN = median)。请注意，n<-3L，而不是 2。
@Miff 非常感谢。有用。虽然我必须在中位数之前更改 i+1 的位置以包括 9

标签： r median

【解决方案1】：

这里有三个选项。
首先是数据。

x <- 1:10
y <- data.frame(x)
n <- 2L

1 基础 R，for 循环。

median_roll <- numeric(length(x) - n)
for(i in seq_along(median_roll)){
  median_roll[i] <- median(y[i:(i+n), ])
}

median_roll
#[1] 2 3 4 5 6 7 8 9

2基R，sapply循环。

sapply(seq_along(median_roll), \(i, n) median(y[i:(i+n), ]), n = 2L)
#[1] 2 3 4 5 6 7 8 9

3包zoo。

zoo::rollapplyr(y$x, width = 3L, FUN = median)
#[1] 2 3 4 5 6 7 8 9

【讨论】：

感谢您的详细解答。我需要根据@Miff 编辑我的代码并更改 i+1 的位置。但我也接受你的回答

【解决方案2】：

在 R 中，索引从 1 开始，i 应该相应地初始化。

i <- 1; x <- 1:10; r <- NULL; n <- 2
while (i - 1 + n < length(x)) {
  r[i] <- median(x[i:(i + n)])
  i <- i + 1
}
r
# [1] 2 3 4 5 6 7 8 9

您还可以使用repeat 循环，它可能更具可读性。

i <- 1; x <- 1:10; r <- NULL; n <- 2
repeat {
  r[i] <- median(x[i:(i + n)])
  i <- i + 1
  if (i - 1 + n == length(x))
    break
}
r
# [1] 2 3 4 5 6 7 8 9

sapply(Map()) 解决方案。

x <- 1:10; n <- 2
sapply(Map(`+`, list(1:(n + 1L)), seq(0, length(x) - n - 1L, n - 1L)), 
       \(s) median(x[s]))
# [1] 2 3 4 5 6 7 8 9

【讨论】：