将pmap列表参数应用于嵌套在另一个函数中的函数

时间:2019-09-05 18:28:39

标签: r purrr pmap

我需要借助pmap变体执行一些按行操作,但是在将参数列表(即“ .l”参数)传递给嵌套在另一个函数中。

我尝试了各种方法,包括传递参数名称和点对点语法都无济于事。我需要知道是否有一种方法可以将其扩展到更复杂的功能。

假设我有以下数据框,并且我想粘贴每行的前两列。我可以使用以下代码轻松做到这一点:

dff <- data_frame(
  first  = c("A", "B"),
  second = c("X", "Y"),
  third  = c("L", "M")
)

df_easy <- dff %>% 
  mutate(joined_upper = pmap_chr(list(first, second), function(...) paste(..., sep = "&")))

df_easy
#> # A tibble: 2 x 4
#>   first second third joined_upper
#>   <chr> <chr>  <chr> <chr>       
#> 1 A     X      L     A&X         
#> 2 B     Y      M     B&Y

但是,如果我想扩展它以便在合并前两个字母时将其小写,则尝试失败。我想看看是否可以得到dff3。

# df_hard <- dff %>% 
#   mutate(joined_smaller = pmap_chr(list(first, second), function(...) paste(tolower(...), sep = "&")))

dff3 <- data.frame(
  first  = c("A", "B"),
  second = c("X", "Y"),
  third  = c("L", "M"),
  joined_smaller = c("a&X", "b&Y")
)

dff3
#>   first second third joined_smaller
#> 1     A      X     L            a&X
#> 2     B      Y     M            b&Y

1 个答案:

答案 0 :(得分:2)

这里是一种选择。请注意,pastestr_c是矢量化的

library(dplyr)
library(stringr)
dff %>% 
     mutate(joined_sma = str_c(tolower(first), second, sep="&"))

并假设这只是pmap的练习

library(purrr)    
dff %>%
   mutate(joined_sma = pmap_chr(list(first, second), ~ c(...) %>% 
                {str_c(tolower(first(.)), .[-1], sep="&")}
      ))
# A tibble: 2 x 4
# first second third joined_sma
#  <chr> <chr>  <chr> <chr>     
#1 A     X      L     a&X       
#2 B     Y      M     b&Y       

此外,由于只有两列,我们可以使用约定.x.y来调用这些

dff %>%
   mutate(joined_sma = pmap_chr(list(first, second), ~     
       str_c(tolower(.x), .y, sep="&")
  ))

注意:在这里,我们使用的是str_c而不是paste,因为当缺少值(NA)时,它可能具有不同的行为