如何在r中给出对角线和非对角线元素的矩阵?

时间:2014-04-13 08:44:14

标签: r matrix fill diagonal

我有一个矩阵的元素如下:

diag= rep(1,5)
offdiag = c(rep(1:4), rep(1:3), rep(1:2), 1)

我想要的最终矩阵应该是一个对称矩阵,如下所示:

1 1 2 3 4
1 1 1 2 3 
2 1 1 1 2
3 2 1 1 1
4 3 2 1 1

其中对角线由diag填充,而下三角区域由offdiag柱填充。

在实践中,所有数字都是随机的。所以我需要一种通用的方法来用元素填充矩阵。

提前致谢!

1 个答案:

答案 0 :(得分:5)

试试这个:

m <- matrix(NA, ncol = length(diag), nrow = length(diag))
m[lower.tri(m)] <- offdiag
m[upper.tri(m)] <- t(m)[upper.tri(t(m))]
diag(m) <- diag
m
#      [,1] [,2] [,3] [,4] [,5]
# [1,]    1    1    2    3    4
# [2,]    1    1    1    2    3
# [3,]    2    1    1    1    2
# [4,]    3    2    1    1    1
# [5,]    4    3    2    1    1

另一种选择:手动创建距离矩阵并从那里开始工作。

class(offdiag) <- "dist"
attr(offdiag, "Size") <- length(diag)
out <- as.matrix(offdiag)
diag(out) <- diag
out
相关问题