计算一系列行的平均值

Question

给定一个这样的数据框：

A <- c(1,2,3,4,NA,6,7,8,9,10,11,12,13,14,15)
B <- c(NA,NA,NA,20,NA,NA,NA,15,NA,NA,NA,NA,11,NA,9)
DF <- data.frame(A, B)

我想根据 B 列中的值计算 A 列中一系列值的平均值。具体来说，每当 B 列中有一个非 NA 值时，我想计算A 列上方第 2 行和下方第 2 行范围的平均值。

例如，B 列的第一个非 NA 值是 20。所以我想计算上面两行 (2, 3)、下面两行 (NA, 6) 和相邻行 (4)。所以：

mean(2,3,4,NA,6)

同样，B 行中的下一个非 NA 值是 15。这将是

mean(6,7,8,9,10)

因此，整个数据框的最终结果将是一个新列 C

DF$C <- c(NA,NA,NA,3.75,NA,NA,NA,8,NA,NA,NA,NA,13,NA,14)

Answer 1

这是 zoo 包的一种方法：

library(zoo)
width <- 5   # the observation ± 2

DF$C <- rollapply(DF$A, width, mean, na.rm = TRUE, partial = TRUE)

# when DF$B is NA, assign NA to corresponding DF$C
DF$C[is.na(DF$B)] <- NA

partial = TRUE 允许在无法容纳整个 window 的 DF$A 向量的前导和尾随部分使用部分 window 计算平均值（即 DF$A 的前 2 个和后 2 个值，其中大小 5 的 window 是不可能的）。

Answer 2

您可以尝试以下方法。

nona <- !is.na(DF$B)
DF$C <- replace(
    DF$B, 
    nona, 
    vapply(which(nona), function(i) {
        ii <- (i-2):(i+2)
        mean(DF$A[ii[ii > 0]], na.rm = TRUE)
    }, 1)
)

这里我们在 B 列中找到非 NA 值，然后使用该向量为我们想要在 A 列中找到平均值的值设置索引，注意删除任何可能的负下标如果 B 列的前一个或两个值不是 NA，则会发生。上面的代码给出了 DF.

的以下结果

    A  B     C
1   1 NA    NA
2   2 NA    NA
3   3 NA    NA
4   4 20  3.75
5  NA NA    NA
6   6 NA    NA
7   7 NA    NA
8   8 15  8.00
9   9 NA    NA
10 10 NA    NA
11 11 NA    NA
12 12 NA    NA
13 13 11 13.00
14 14 NA    NA
15 15  9 14.00

计算一系列行的平均值

Calculate mean of a range of rows

r

mean

dataframe