添加，使用R的矩阵的对角线(Adding Diagonals of a matrix using R

我想补充了上部对角线从中间开始的矩阵，在列递增，直到（1，N），n是最后一列，并节省每对角线的每个总和。 我的代码只添加中间对角线，我怎样才能通过矩阵环获得对角线的总和

A <- matrix(c(2, 4, 3, 1,
             5, 7, 1, 2,
             3, 2, 3, 4, 
             1, 5, 6, 0), # the data elements 
    nrow = 4, # number of rows 
    ncol = 4, # number of columns 
    byrow = TRUE) # fill matrix by rows

sum <- 0
print(A)
for (a in 1){
  for (b in 1:ncol){
    if (a<-b){
      sum = sum + A[a,b]
      print (sum) 
    }
  }
}

这里是我的结果

> print(A)
     [,1] [,2] [,3] [,4]
[1,]    2    4    3    1
[2,]    5    7    1    2
[3,]    3    2    3    4
[4,]    1    5    6    0

for (a in 1){
  for (b in 1:ncol){ 
    if (a<-b){
      sum = sum + A[a,b]
      tail(sum, n=1)
    }
  }
}


12

Answer 1:

你需要diag提取所有主对角线元素sum ，让他们的总和

sum(diag(A))

我不知道你问什么，但如果你也想提取上三角矩阵，可以使用A[upper.tri(A)]不包括主对角线元素，还可以设置diag=TRUE将它们包括A[upper.tri(A, diag = TRUE)]

@shegzter根据您的意见，您可以使用col和row与逻辑比较相结合==得到你想要的数字。

> A[row(A)==col(A)] # this gives the same out put as `diag(A)`
[1] 2 7 3 0
> A[row(A)+1==col(A)]
[1] 4 1 4
> A[row(A)+2==col(A)]
[1] 3 2
> A[row(A)+3==col(A)]
[1] 1

如果你希望他们每个人的总和，所以使用sum对这些元素：

> sum(A[row(A)==col(A)])
[1] 12
> sum(A[row(A)+1==col(A)])
[1] 9
> sum(A[row(A)+2==col(A)])
[1] 5
> sum(A[row(A)+3==col(A)])
[1] 1

如果你的目标是获得以下和12 + 9 + 5 + 1，那么你可以用它做一次全部upper.tri和sum

> sum(A[upper.tri(A, diag = TRUE)])
[1] 27

或者没有对角线元素：

> sum(A[upper.tri(A)])
[1] 15

Answer 2:

下面返回每个对角线的总和：

sapply(split(A, col(A) - row(A)), sum)
# -3 -2 -1  0  1  2  3 
#  1  8 13 12  9  5  1

因此，只得到了上，那些你可以使用

tail(sapply(split(A, col(A) - row(A)), sum), ncol(A))
#  0  1  2  3 
# 12  9  5  1 

使用的缺点tail的是，我们还计算斜下方的款项。 因此，为了节省一些时间，当A大的，你可能想使用

sapply(split(A[upper.tri(A, diag = TRUE)], (col(A) - row(A))[upper.tri(A, diag = TRUE)]), sum)
#  0  1  2  3 
# 12  9  5  1 

文章来源: Adding Diagonals of a matrix using R