将列表中的所有矩阵转换为数据框在R中。

6

我在R中使用lme函数并得到了输出结果。

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

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

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


你期望的输出是什么?也许是 broom::tidy(fm2) - akrun
输出是相同的对象(列表),但所有矩阵都转换为数据框。 - MyQ
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)

1
class(lapply(fm2, function(x) {if(any(class(x)=="matrix")) as.data.frame(x) else x})$varFix) [1] "data.frame" class(fm2$varFix) "matrix" 不错的解决方案! - Erdogan CEVHER

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接