Question

我一直在尝试编写一个接受矩阵（x）和向量（cut.vec）的函数并输出一个列表，其中列表的每个元素都是输入矩阵中某些列的组合。输入向量中的每个元素都是想要对矩阵进行分区的索引。然后我想将每个分区保存到列表中的元素并返回该列表。

这是我到目前为止所得到的：

这是我正在运行的实际功能：

make.cut <- function(x, cut.vec){

    ran.once <- 0 #This checks for first run
    out <- list() #This creates the output list
    temp.matrix <- matrix() #For holding data

    for(i in 1:length(cut.vec)){

        for(n in 1:cut.vec[i]){

            if(cut.vec[i]<n){
                #Do nothing
            }else{
                hold <- x[,n]
                if(ran.once != 0){
                    temp.matrix <- cbind(temp.matrix, hold)
                }else{
                    temp.matrix <- hold
                    ran.once <- 1
                }
            }
        }

        out[[i]] <- temp.matrix
        temp.matrix <- matrix()
    }
    return(out)
}

当我运行这个时，我得到一个列表，但只有第一个元素是正确的。除第一个元素之外的每个元素仅包含输入矩阵的一列。

**Example Input**
x<-matrix(c(341, 435, 834, 412, 245, 532.2, 683.4, 204.2, 562.7, 721.5, 149, 356, 112, 253, 211, 53, 92, 61, 84, 69), nrow=4)

x= 341   435   834   412   245
   532.2 683.4 204.2 562.7 721.5
   149   356   112   253   211
   53    92    61    84    69

cut.vec = c(2, 3, 5)

out <- make.cut(x, cut.vec):

a <- out[[1]]
b <- out[[2]]
c <- out[[3]]

**Intended Output**
a= 341   435  
   532.2 683.4 
   149   356   
   53    92    

b= 834  
   204.2
   112  
   61   

c= 412   245
   562.7 721.5
   253   211
   84    69

**Actual Output**
a= 341   435   
   532.2 683.4 
   149   356  
   53    92   

b= 435   
   683.4 
   356   
   92    

c= 834  
   204.2
   112  
   61

我可以手动从控制台执行此操作，一次一个元素并且可以正常工作，但每次尝试使用make.cut函数执行此操作时都会中断。

这是我在终端手工完成的方式：

cut.vec<-c(3, 5)

a<-x[,1]
b<-x[,2]
c<-x[,3]

temp <- cbind(a,b,c)

out[[1]] <- temp

cut.vec [2]等于5

a<-x[,4]
b<-x[,5]

temp <- cbind(a,b)

out[[2]] <- temp

但是，当我尝试在函数中应用相同的方法时，它会中断。

Answer 1

您可以使用以下方法沿矢量“剪切”矩阵：

示例数据：

mat <- matrix(1:16, nrow = 2)
#      [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8]
# [1,]    1    3    5    7    9   11   13   15
# [2,]    2    4    6    8   10   12   14   16

cutvec <- c(2,5)

首先，在mat上删除cutvec的列号：

cuts <- cut(seq(ncol(mat)), c(0, cutvec - 1))

然后，您可以使用tapply：

创建包含子集的列表

tapply(seq(ncol(mat)), cuts, function(x) mat[, x, drop = FALSE])

# $`(0,1]`
#      [,1]
# [1,]    1
# [2,]    2
#
# $`(1,4]`
#      [,1] [,2] [,3]
# [1,]    3    5    7
# [2,]    4    6    8

将矩阵划分为R中的列表时的奇怪输出

1 个答案: