连续一些位置的总和 - R.

Question

考虑通用矩阵nxn - 例如:

A <- matrix(1:16, nrow = 4, ncol = 4)

如何计算"右下"三角形中的行总和并在向量中显示信息？

Answer 1

找到封装所需数据的掩码:

> mask <- apply(lower.tri(A, diag = FALSE), 1, rev)
> mask
      [,1]  [,2]  [,3]  [,4]
[1,] FALSE FALSE FALSE FALSE
[2,] FALSE FALSE FALSE  TRUE
[3,] FALSE FALSE  TRUE  TRUE
[4,] FALSE  TRUE  TRUE  TRUE

将此掩码相乘并计算总和:

> A * mask
     [,1] [,2] [,3] [,4]
[1,]    0    0    0    0
[2,]    0    0    0   14
[3,]    0    0   11   15
[4,]    0    8   12   16

>  rowSums(A * mask)
[1]  0 14 26 36

Answer 2

这是一个解决方案。关键是使用lower.triorupper.tri函数，然后使用orapply对每一行或每一列进行排序。最后计算每一行的总和。

# Create example data frame
m <- matrix(1:16, 4)
m
#      [,1] [,2] [,3] [,4]
# [1,]    1    5    9   13
# [2,]    2    6   10   14
# [3,]    3    7   11   15
# [4,]    4    8   12   16

# Calculate the sum 
rowSums(m * apply(lower.tri(m), 1, sort))
# [1] 0 14 26 36

rowSums(m * apply(upper.tri(m), 2, sort))
# [1] 0 14 26 36