Question

我正试图从矢量中创建这样的矩阵：

    vec= c(2, 5, 9)
    > A
            [,1] [,2] [,3] [,4]
    [1,]     2    0    0    0
    [2,]     5    3    0    0
    [3,]     9    7    4    0

实际上，第一列始终是向量元素，第二列以0开始，然后是（5-2 = 3），然后第二列的thirld元素是（9-2 = 7）。然后第三列从0开始，然后是0和（9-5 = 4），最后一列始终为零。可能是vec更改为任何数字的长度，例如4,5，......。如何编写有效的函数或代码来创建此矩阵？

Answer 1

我不知道效率，但这里有两个不使用for循环的解决方案：

n <- length(vec)    
A <- replicate(n+1, vec) - cbind(0, t(replicate(n, vec)))
A[upper.tri(A)] <- 0

这个更长但只创建一个矩阵

n <- length(vec)
A <- replicate(n, vec)
A <- A - t(A)
A <- cbind(vec, A)
A[upper.tri(A)] <- 0

Answer 2

我认为这会做你想做的事情：

f = function(vec)
{
   n = length(vec)  
   M = matrix(0,n,n+1)
   M[,1] = vec
   for(i in 1:n) M[,i+1] = c(rep(0,i),vec[-c(1:i)]-vec[i])
   return(M)
}

vec = c(2,5,9)
f(vec)
     [,1] [,2] [,3] [,4]
[1,]    2    0    0    0
[2,]    5    3    0    0
[3,]    9    7    4    0

在R中创建矩阵的有效方法

2 个答案: