重新创建 diag() 函数

Question

我目前正在尝试复制 R 中的 diag() 函数以获得更好的理解。不幸的是，我的解决方案仅对对称矩阵有用。

diagnew <- function(x){
  k <- c()
  for(i in 1:nrow(x)){
    k[i] <- x[i,i]
  }
  return(k)
}

如果有任何提示可以概括我的函数，我将不胜感激。

Answer 1

错误来自循环传递 min(nrow, ncol)。

diagnew <- function(x){
  n <- min(nrow(x), ncol(x))
  k <- vector(class(x[1,1]), n)
  for(i in seq.int(n)){
    k[i] <- x[i,i]
  }
  k
}

mat <- matrix(1:10, 5)
diag(mat)
diagnew(mat)

identical(diag(mat), diagnew(mat))
#[1] TRUE