制作一个矩阵列表，其中每个矩阵是一行数据框

问题描述

我想制作一个 2x2 矩阵的列表，其中数据帧的每一行都变成一个 2x2 矩阵。我将使用一个巨大的数据集（为了方便起见，玩具示例），所以我试图弄清楚如何在没有 for 循环的情况下做到这一点。我一直在尝试使用自制函数和 apply() 来执行此操作，但收到一条错误消息。

dat<-data.frame(x1=c(2,3,4),x2=c(1,2,3),x3=c(5,76,7),x4=c(4,6,0))

make_matrix<-function(df){
  with(dat,matrix(c(x1,x2,x3,x4),byrow=T,nrow=2,ncol=2))
  }

apply(dat,1,FUN=make_matrix)
  
Error in eval(substitute(expr),data,enclos = parent.frame()) : 
  numeric 'envir' arg not of length one 

解决方法

我们可以使用 asplit

lapply(asplit(dat,1),matrix,2,byrow = TRUE)

-输出

#[[1]]
#     [,1] [,2]
#[1,]    2    1
#[2,]    5    4

#[[2]]
#     [,]    3    2
#[2,]   76    6

#[[3]]
#     [,]    4    3
#[2,]    7    0

---

OP 的函数参数和主体不匹配，即它采用参数 'df' 并且在函数内部使用 dat。但是，这不是唯一的问题。当我们遍历该行时，它是一个值向量，with 对这些值不起作用。因为我们需要一个 list，所以 matrix 可以被包裹在一个 list

make_matrix <- function(x) list(matrix(x,byrow = TRUE,ncol = 2,nrow = 2))

应该可以

do.call(c,apply(dat,1,make_matrix))
#[[1]]
#     [,]    7    0

,

这是另一个使用 array

的选项

mat <- array(c(t(dat)),c(2,3))
mat[] <- apply(mat,3,t)

给出

> mat,1

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

     [,]    3    2
[2,]   76    6,3

     [,]    4    3
[2,]    7    0

---

如果你想得到list形式的结果，你可以试试

Map(t,asplit(array(c(t(dat)),3)),3))

给出

[[1]]
     [,]    5    4

[[2]]
     [,]   76    6

[[3]]
     [,]    7    0