将列表中的所有矩阵转换为R中的data.frames

问题描述 投票:6回答:1

我在lmeR函数的输出。

library(nlme)
fm2 <- lme(distance ~ age + Sex, data = Orthodont, random = ~ 1)
str(fm2)

如您所见,输出的一些元素(fm2)是矩阵,例如: fm2$varFix

我正在寻找一个函数来接受一个对象并将所有子矩阵转换为data.frames

r list dataframe matrix
1个回答
6
投票

也许是这样的:

lapply(fm2, function(x) {if(any(class(x)=="matrix")) as.data.frame(x) else x})

编辑:虽然这个答案已被接受,但我对解决方案不满意,因为它不会递归地转换列表中的矩阵,并且元素的属性会丢失。我相信以下解决方案可以解决这两个问题:

library(nlme)
fm2 <- lme(distance ~ age + Sex, data = Orthodont, random = ~ 1)
str(fm2)

replace_sub_dataframes <- function(x)
{
  if(any(class(x)=="list"))
  {
    x_copy = x
    attrs = setdiff(names(attributes(x)),"names")
    x = lapply(x,replace_sub_dataframes)
    if(length(attrs)>0)
    {
      for(i in 1:length(attrs))
      {
        attr(x,attrs[i]) <- replace_sub_dataframes(attr(x_copy,attrs[i]))
      }
    }
    return(x)
  }
  else
  {
    if(any(class(x)=="matrix")) 
      return(as.data.frame(x)) 
    else 
      return(x)
  }
}

fm3 = lapply(fm2, replace_sub_dataframes)
© www.soinside.com 2019 - 2024. All rights reserved.