R拆分矩阵并合并它们

时间:2014-08-16 09:05:17

标签: r split

我正在寻找一个更优雅的R代码来执行以下操作:

mat <- matrix(data = c(1,1,0,0,1,1,2,2,2,0,0,0,0,1,1,2,2,2), nrow = 3)
colnames(mat) <- c("x1","x2","x3","x1","x2","x3")
len <- 2
ncol.mat <- 1:NCOL(mat)

split.vec <- split(ncol.mat, cut(ncol.mat, breaks = len))
split.vec <- as.data.frame(split.vec)

new.mat <- data.frame()

for(i in 1:len){
  vv <- mat[,split.vec[,i]]
  new.mat <- rbind(new.mat,vv)
}

new.mat

有没有人有一个很好的方法来做到这一点?我还没有找到一个优雅的解决方案,但我确信有一个非常简单的解决方案。谢谢你的帮助。

1 个答案:

答案 0 :(得分:1)

尝试:

  apply(array(mat, dim=c(3,3,2)),2, c)
  #     [,1] [,2] [,3]
  #[1,]    1    0    2
  #[2,]    1    1    2
  #[3,]    0    1    2
  #[4,]    0    0    2
  #[5,]    0    1    2
  #[6,]    0    1    2

或者

  m2 <- aperm(array(mat, dim=c(3,3,2)), c(1,3,2))
  dim(m2) <- c(prod(dim(m2)[-1]), dim(m2)[1])
   m2
  #      [,1] [,2] [,3]
  # [1,]    1    0    2
  # [2,]    1    1    2
  # [3,]    0    1    2
  # [4,]    0    0    2
  # [5,]    0    1    2
  # [6,]    0    1    2