4

当我有一个包含两个输入列(a 和 b)的列表列数据框(df1)时,我可以使用 map2 获得一个新列

d1 <- mtcars %>% slice(1:10) %>% group_by(cyl) %>% nest(.key = "a")
d2 <- mtcars %>% slice(11:20) %>% group_by(cyl) %>% nest(.key = "b")
df1 <- d1 %>% left_join(d2) 

my_fun <- function(a, b) {
  # some manipulations that involve the dataframe a and b and result in 
  # another dataframe
  # below, just a trivial manipulation 
  a %>% bind_rows(b)  
}

df1 %>% mutate(z = map2(a, b, my_fun)) 

现在假设我有一个带有 3 个输入列(a、b 和 c)的新数据框(df2)。

d3 <- mtcars %>% slice(21:10) %>% group_by(cyl) %>% nest(.key = "c")
df2 <- d1 %>% left_join(d2) %>% left_join(d3)

如何使用 3 个输入列获取新列?由于没有map3,我想我应该使用pmap,但我不知道如何。

4

1 回答 1

4

正如@Sathish 所评论的那样pmap可以使用,这可以mutate通过将列放在 a 中list然后应用函数来使用

library(tidyverse)
df2 %>% 
   mutate(z = pmap(list(a, b, c), bind_rows))
于 2018-06-20T14:29:44.743 回答