r语言 - 基于谓词函数 (dplyr::mutate_if) 改变数据框的列



我想使用 dplyr 的 mutate_if() 函数将列表列转换为数据框列,但是当我尝试这样做时遇到了一个令人费解的错误。我正在使用 dplyr 0.5.0、咕噜声 0.2.2、R 3.3.0。

基本设置如下所示: 我有一个数据框d,其中一些列是列表:

d <- dplyr::data_frame(
  A = list(
    list(list(x = "a", y = 1), list(x = "b", y = 2)),
    list(list(x = "c", y = 3), list(x = "d", y = 4))
  ),
  B = LETTERS[1:2]
)

我想使用以下函数将列表列(在本例中为 d$A)转换为数据框列:

tblfy <- function(x) {
  x %>%
    purrr::transpose() %>%
    purrr::simplify_all() %>%
    dplyr::as_data_frame()
}

也就是说,我希望列表列d$A替换为列表lapply(d$A, tblfy),即

[[1]]
#  A tibble: 2 x 2
      x     y
  <chr> <dbl>
1     a     1
2     b     2
[[2]]
# A tibble: 2 x 2
      x     y
  <chr> <dbl>
1     c     3
2     d     4

当然,在这个简单的情况下,我可以做一个简单的重新分配。然而,关键是我想以编程方式做到这一点,最好是使用 dplyr,以一种普遍适用的方式处理任何数量的列表列。

这是我偶然发现的地方:当我尝试使用以下应用程序将列表列转换为数据框列时

d %>% dplyr::mutate_if(is.list, funs(tblfy))

我收到一条错误消息,我不知道如何解释:

Error: Each variable must be named.
Problem variables: 1, 2

为什么mutate_if()会失败?如何正确应用它以获得所需的结果?

备注

一位评论者指出,函数tblfy()应该矢量化。这是一个合理的建议。但是——除非我错误地矢量化——这似乎并不能找到问题的根源。插入矢量化版本的tblfy()

tblfy_vec <- Vectorize(tblfy)

进入mutate_if()失败并显示错误

Error: wrong result size (4), expected 2 or 1

更新

在获得一些咕噜声的经验之后,我现在发现以下方法很自然,尽管有些冗长:

d %>%
  map_if(is.list, ~ map(., ~ map_df(., identity))) %>%
  as_data_frame()

这或多或少与下面的@alistaire解决方案相同,但使用 map_if() ,或者。 map(),代替mutate_if(),或。 Vectorize() .

原始的tblfy函数对我来说是错误的(即使它的元素是直接链接的),所以让我们重建一下它,添加矢量化,这让我们避免了不必要的先前rowwise()调用:

tblfy <- Vectorize(function(x){x %>% purrr::map_df(identity) %>% list()})

现在我们可以很好地使用mutate_if

d %>% mutate_if(purrr::is_list, tblfy)
## Source: local data frame [2 x 2]
## 
##                A     B
##           <list> <chr>
## 1 <tbl_df [2,2]>     A
## 2 <tbl_df [2,2]>     B

。如果我们解开巢穴看看那里有什么,

d %>% mutate_if(purrr::is_list, tblfy) %>% tidyr::unnest()
## Source: local data frame [4 x 3]
## 
##       B     x     y
##   <chr> <chr> <dbl>
## 1     A     a     1
## 2     A     b     2
## 3     B     c     3
## 4     B     d     4

几点说明:

  • map_df(identity)似乎比任何替代配方更有效地构建 tibble。我知道identity电话似乎没有必要,但大多数其他一切都会中断。
  • 我不确定tblfy有多大用处,因为它在某种程度上取决于列表列中列表的结构,这可能会有很大差异。不过,如果你有很多类似的结构,我想它是有用的。
  • 可能有一种方法可以用pmap而不是Vectorize来做到这一点,但我无法通过一些粗略的尝试让它工作。

无需任何复制的就地转换:

library(data.table)
for (col in d) if (is.list(col)) lapply(col, setDF)
d
#Source: local data frame [2 x 2]
#
#                A B
#1 <S3:data.frame> A
#2 <S3:data.frame> B

最新更新