我有这样的小标题:
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)
答案 0 :(得分:1)
我们可以使用unnest
library(tidyverse)
res %>%
select(-lists) %>%
unnest