1

我必须将向量转换为矩阵

x<-c(1:5)
mat<-matrix(nrow=3, ncol=3)
for (i in 1:3){
  for (j in 1:3){
    if (i==j) {
      mat[i,j]<-x[3]
    } else 
      if (i < j) { ##for upper diagonal
        mat[i,j]<-x[j]
      }
  }
}

结果矩阵应为

    [,1]  [,2] [,3]
[1,] 1      2    3
[2,] 4      1    2
[3,] 5      4    1

我知道它是一种 toeplitz 矩阵,并且 R 中有可用的包,但我必须使用嵌套的 for 循环来做到这一点。

4

1 回答 1

4

这可以完成,toeplitz并且功能来自base Rie 不需要包

 `diag<-`(toeplitz(x)[-(2:3), 1:3], 1)
     [,1] [,2] [,3]
[1,]    1    2    3
[2,]    4    1    2
[3,]    5    4    1

关于嵌套循环,一个选项是

x <- c(1:5)
mat <- matrix(nrow=3, ncol=3)
for (i in 1:3){
  for (j in 1:3){
    if (i==j) {
      mat[i,j]<-x[1]
    } else if (i > j) { 
         if((i + j) < length(x)){
            mat[i,j] <- x[(i + j + 1)]
         } else {
            mat[i, j] <- x[(i + j) - 1]
         }  
      } else {
          if((i + j) < length(x)) {
           mat[i, j] <- x[j]
          } else {
            mat[i, j] <- x[i]
          }
              }
  }
}

-输出

> mat
     [,1] [,2] [,3]
[1,]    1    2    3
[2,]    4    1    2
[3,]    5    4    1
于 2021-10-12T17:35:26.893 回答