r - 如何在对角线中遍历矩阵并返回每个位置的索引？

Question

有一个矩阵 N x N，我想在对角线中遍历这个矩阵并返回索引位置。

如果我有一个 4x4 矩阵，代码应该返回 (1,1)； (1,2); (2,1); (1,3); (2,2); (3,1); (1,4); (2,3); (3,2); (4,1);等等

我正在尝试在 R Studio 中执行此操作

Answer 1

1) row(m) + col(m) 沿反向对角线保持不变，并且在我们按行排序的反向对角线内:

m <- matrix(1:16, 4, 4) # test matrix

m[order(row(m) + col(m), row(m))]
## [1]  1  5  2  9  6  3 13 10  7  4 14 11  8 15 12 16

2) 不像 (1) 那样紧凑，但这里有一个变体，它使用相同的原理，但使用 outer 和回收而不是 row 和 col:

k <- nrow(m)
m[ order(outer(1:k, 1:k, "+") + 0:(k-1)/k) ]
## [1]  1  5  2  9  6  3 13 10  7  4 14 11  8 15 12 16