R中的行滑动窗口

Question

我有一个数据框，它有大约 50000 行，我想循环它，每个循环移动 10 行直到它到达 50000 行，这是代码。

for (i in 1:50000){

  if(df[[1]][i,8] > 1){

    hom <- hom + df[[1]][i,11]
  }else{
    het<-het + het[[1]][i,11]
  }
  score[i] = hom/(hom+het)

}

在代码中，我正在检查第 8 列中的条件是真还是假，以便应用适当的函数。 然后我得到分数。 所以我基本上想得到每行10行的分数{1-10,2-11,3-12,4-13........49990-50000}

如果你能帮忙，我会很高兴 谢谢

Answer 1

以下代码实现了一个 R 函数，该函数接收一个向量和一个表示任意大小的滑动窗口的整数作为输入。该函数输出一个向量，其中包含窗口每个缩管内元素的平均值（您可以调整代码以进行任何其他类型的计算）。

# Function that computes an sliding window based on the mean values
compute_mean_feat <- function(vector, window) {
  mean <- c()
  for (i in 1:length(vector)) {
    if (i <= window) {
      sum <- 0
      for (k in 1:i) {
        sum  <- vector[k] + sum
      }
      mean <- c(mean, as.numeric(sum / i))
    } else{
      sum <- 0
      for (j in (i - window + 1):i) 
        sum  <- vector[j] + sum
      }
      mean <- c(mean, as.numeric(sum / i))
    }
  }
  return(mean)
}

您可以看到将滑动窗口为 10 的 compute_mean_feat 函数应用于虚拟 vector 的结果。

# Manual example for testing
window <- 10    
vector <- c(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
compute_mean_feat(vector, window)

对于数据框，您只需将此函数应用于所需的列，它将为您提供其行的滑动窗口向量。