Question

道歉可能是一个非常简单的问题。

我是新手在R中使用purrr包而我正在努力尝试将第二个参数传递给函数。

library(dplyr)
library(purrr)

my_function <- function(x, y = 2) {
  z = x + y
  return(z)
}

my_df_2 <- my_df %>%
  mutate(new_col = map_dbl(.x = old_col, .f = my_function))

这是有效的，而且大多数情况下我不需要改变y的值，但是如果我必须通过mutate＆amp;传递一个不同的y值（比如y = 3）。地图组合，它的语法是什么？

非常感谢你！

Answer 1

我认为您需要做的就是修改map_dbl，如下所示：

library(dplyr)
library(purrr)

df <- data.frame(a = c(2, 3, 4, 5.5))

my_function <- function(x, y = 2) {
  z = x + y
  return(z)
}

df %>%
  mutate(new_col = map_dbl(.x = a, y = 3, .f = my_function))

    a new_col
1 2.0     5.0
2 3.0     6.0
3 4.0     7.0
4 5.5     8.5

Answer 2

另一个想法是使用以下语法。

library(dplyr)
library(purrr)

# The function
my_function <- function(x, y = 2) {
  z = x + y
  return(z)
}

# Example data frame
my_df <- data_frame(old_col = 1:5)

# Apply the function   
my_df_2 <- my_df %>%
  mutate(new_col = map_dbl(old_col, ~my_function(.x, y = 3)))

my_df_2
# # A tibble: 5 x 2
# old_col new_col
#     <int>   <dbl>
# 1       1      4.
# 2       2      5.
# 3       3      6.
# 4       4      7.
# 5       5      8.

如何在R中使用purrr包的map函数时将第二个参数传递给函数

2 个答案: