地图功能中的第二个参数

时间:2018-07-09 09:25:48

标签: r dplyr purrr

我有这样的小标题:

uuu <- structure(list(Id = c("a", "b"),
                      lists = list(c(1:2), c(100:103))
),
.Names = c("Id", "lists"),
row.names = c(NA, 2L), class = c("tbl_df", "tbl", "data.frame"))

> uuu
# A tibble: 2 x 2
  Id    lists    
* <chr> <list>   
1 a     <int [2]>
2 b     <int [4]>

在此示例中,它有2行和2列,第一列是称为id的字符列,第二列是lists并包含整数列表。

现在,我想在列表元素上映射一个函数,例如sqrt函数:

res <- uuu %>% mutate(res = map(lists, sqrt))

> res
# A tibble: 2 x 3
  Id    lists     res      
  <chr> <list>    <list>   
1 a     <int [2]> <dbl [2]>
2 b     <int [4]> <dbl [4]>

res现在有了一个新列,其中包含sqrt函数的所有结果。

现在,我想在res列中取消列出结果,但是我还想拥有一个新列,以指示结果来自哪个id

我可以取消列出这样的结果:

> unlist(res$res)
[1]  1.000000  1.414214 10.000000 10.049876 10.099505 10.148892

但是我真正想要的是这样的东西(这里是手动进行说明):

res2 <- data.frame(res = unlist(res$res),
            ids = c(rep("a", 2), rep("b", 4)))

> res2
        res ids
1  1.000000   a
2  1.414214   a
3 10.000000   b
4 10.049876   b
5 10.099505   b
6 10.148892   b

理想情况下,对于我定义自己的函数的情况,我希望尽可能在映射部分中直接传递它,例如:

res <- uuu %>% mutate(res = map(lists, my_sqrt, id = id))

my_sqrt如下:

my_sqrt <- function(x, id) return(sqrt(x), id)

1 个答案:

答案 0 :(得分:1)

我们可以使用unnest

library(tidyverse)
res %>% 
  select(-lists) %>%
  unnest