r - R 中的 diag() 函数

Question

有没有办法使用diag()矩阵中的函数而不使用内置函数或迭代？

   M<-matrix(1:9, ncol=3) # make a matrix 

    q5b<-function(M){ #function

    }

我知道 M[1,1] , M[2,2] , 和 M[3,3]会给我与 diag(M) 相同的输出.但是，我想不出没有 for 循环的方法。

我的思考过程是我应该有一个条件，其中 row index == column index在矩阵中然后打印该值。我很感激任何建议。

Answer 1

您可以使用功能row和 col查找列号与行号相同的索引:

row(M) == col(M)
#       [,1]  [,2]  [,3]
# [1,]  TRUE FALSE FALSE
# [2,] FALSE  TRUE FALSE
# [3,] FALSE FALSE  TRUE

M[row(M) == col(M)]
# [1] 1 5 9