2012-03-28 114 views
1

我有一个向量或数组操作有点麻烦。R - 矢量/阵列加法

我有三个3D阵列,我想找到它们的平均值。我怎样才能做到这一点?我们不能使用mean(),因为它只返回一个值。

更重要的是一些在阵列中的细胞是NA whic意思是,如果我只是将它们添加像

A = (B + C + D)/3 

的结果将显示NA为好。

我该如何让它识别单元格是否NA然后跳过它。

A = c(NA, 10, 15, 15, NA) 
B = c(10, 15, NA, 22, NA) 
C = c(NA, NA, 20, 26, NA) 

我想的平均输出这些载体是

(10, (10+15)/2, (15+20)/2, (15+22+26)/3, NA) 

我们也不能使用na.omit,因为它会移动索引的顺序。

这是相应的代码。我希望这会有所帮助。

for (yr in 1950:2011) { 
    temp_JFM <- sst5_sst2[,,year5_sst2==yr & (month5_sst2>=1 & month5_sst2<=3)] 
     k = 0 
     jfm=4*k+1 
    for (i in 1:72) { 
     for (j in 1:36) { 
      iposst5_sst2[i,j,jfm] <- (temp_JFM[i,j,1]+temp_JFM[i,j,2]+temp_JFM[i,j,3])/3 
     } 
    }  
} 

那你呢。

它已经被解决了。

纠正它的最简单方法如下所示。

iposst5_sst2[i,j,jfm] <- mean(temp_JFM[i,j,],na.rm=TRUE) 

回答

2

下面是一个例子,这使得这三个值的矢量,这使得na.omit可用:

vectorAverage <- function(A,B,C) { 
    Z <- rep(NA, length(A)) 

    for (i in 1:length(A)) { 
     x <- na.omit(c(A[i],B[i],C[i])) 
     if (length(x) > 0) Z[i] = mean(x) 
    } 
    Z 
} 

在所得:

vectorAverage(A,B,C) 
[1] 10.0 12.5 17.5 21.0 NA 

已编辑:错过了第一个版本输出中的NaN。

7

我不能完全确定你的期望的输出是什么,但我猜你真的想建立的是不是三3D阵列,而是一个四维阵列,然后可以上使用apply

像这样:

#Three 3D arrays... 
A <- array(runif(1:27),dim = c(3,3,3)) 
B <- array(runif(1:27),dim = c(3,3,3)) 
C <- array(runif(1:27),dim = c(3,3,3)) 

#Become one 4D array 
D <- array(c(A,B,C),dim = c(3,3,3,3)) 

#Now we can simply apply the function mean 
# and use it's na.rm = TRUE argument. 
apply(D,1:3,mean,na.rm = TRUE) 
+3

+1 - 同样的,如果你正在处理大数组,你可以使用'applyMe'快得多'rowMeans(D,na.rm = TRUE,dims = 3)' – flodel 2012-03-28 03:56:31