1

我正在尝试用预填充向量 V 填充矩阵 M 的下对角线

我的原始矩阵看起来像这样：

M = matrix(c(.3,.2,.1,0), nrow=4, ncol=5)

M  1  2  3  4 5
1 .3 .3 .3 .3 .3
2 .2 .2 .2 .2 .3
3 .1 .1 .1 .1 .1
4 0  0  0  0   0

我有一个类似的向量：

 V
 .4
 .3
 .25
 .1

现在我想用这个向量填充下三角形，得到：

0  1  2  3  4  5
1 .3 .3 .3 .3 .1
2 .2 .2 .2 .25 .25
3 .1 .1 .3 .3 .3 
4 0  .4 .4 .4 .4

如果我使用该lower.tri函数，它会给出一个错误，所以我构建了一个循环，它只应该从底部填充列：

o <- 5
c <- 2
s <- 1
for(s in (1:o)){  
 for(c in (2:o)){
   M[((o-s):o),c] <- V[1:c]}}

我的想法是像我手动写的那样向上移动：

M[(5-1):5,2] <- V[1:2]
M[(5-2):5,3] <- V[1:3]

最好的方法是什么？

4

2 回答 2

2

您要填充的矩阵部分是原始矩阵的下三角矩阵相对于行方向的反映，因此您可以rev对结果应用一个函数来lower.tri()反转每一行以获得替换元素的索引，并且那么它会很简单：

假设你有矩阵M和向量v：

M
#   X1  X2  X3  X4  X5
#1 0.3 0.3 0.3 0.3 0.3
#2 0.2 0.2 0.2 0.2 0.3
#3 0.1 0.1 0.1 0.1 0.1
#4 0.0 0.0 0.0 0.0 0.0

v
# [1] 0.40 0.30 0.25 0.10

lowerIndex = t(apply(lower.tri(M, diag = TRUE), 1, rev))
M[lowIndex] <- (lowIndex * rev(v))[lowIndex]
M

#   X1  X2  X3   X4   X5
#1 0.3 0.3 0.3 0.30 0.10
#2 0.2 0.2 0.2 0.25 0.25
#3 0.1 0.1 0.3 0.30 0.30
#4 0.0 0.4 0.4 0.40 0.40
于 2016-11-30T22:25:52.573 回答
1

我们可以这样做：

用要插入的 v 的值定义一个完整矩阵

N = matrix(rev(v), 4, 5)

现在我们可以用替换矩阵中的对应值替换 M 中列大于行倒数的元素

R = rev(row(M))
C = col(M)
M[C>R] = N[C>R]

     # [,1] [,2] [,3] [,4] [,5]
# [1,]  0.3  0.3  0.3 0.30 0.10
# [2,]  0.2  0.2  0.2 0.25 0.25
# [3,]  0.1  0.1  0.3 0.30 0.30
# [4,]  0.0  0.4  0.4 0.40 0.40
于 2016-11-30T22:26:54.813 回答