r - 如何将参数传递给引用嵌套数据帧的列名的 purrr:::map ？

Question

我正在使用足球数据的嵌套数据框，并使用适合每个数据框的模型。使用 goalmodel::predict_result，我想根据模型预测每场比赛的结果。 Predict_result 函数需要三个参数:模型、主队向量和客队向量。如何在嵌套数据框中引用列名称？这是我的嵌套数据框的示例:

library(goalmodel)
library(tidyverse)

nested_df
# A tibble: 4 x 3
# Groups:   League [4]
  League                     data model     
  <chr>          <list<df[,133]>> <list>    
1 F1             [380 x 133] <goalmodl>
2 E0             [380 x 133] <goalmodl>
3 SP1            [380 x 133] <goalmodl>
4 D1             [308 x 133] <goalmodl>

如果我想预测这个数据帧的第一个元素(即 F1)的结果，我会写

predict_result(nested_df$model[[1]], team1=nested_df$data[[1]]$HomeTeam, team2=nested_df$data[[1]]$AwayTeam, return_df = TRUE)

它返回所需结果的数据帧。我尝试使用 purrr::map 迭代上述函数:

map(nested_df,~predict_result(.x$model,
                                 team1=.x$data[[.]]$HomeTeam,
                                 team2=.x$data[[.]]$AwayTeam,
                                 return_df = TRUE))

它不起作用，错误显示:

Error in .x$data : $ operator is invalid for atomic vectors

如果有任何帮助和建议，我将不胜感激，提前致谢。 -----以下是再现性的示例:

df <- tibble(League = c("F1","E0","SP1","D1"),
             HomeTeam = c("TeamA","TeamB","TeamC","TeamD"),
             AwayTeam = c("TeamE","TeamF","TeamG","TeamH"),
             FTHG = c(0,1,2,0),
             FTAG = c(0,1,0,2))

nested_df <- df %>%
  group_by(League)%>%
  nest()%>%
  mutate(model = map(data,~goalmodel::goalmodel(goals1 = .x$FTHG, goals2 = .x$FTAG,
                                     team1 = .x$HomeTeam, team2 = .x$AwayTeam,
                                     ,rs=TRUE)))

nested_df
# A tibble: 4 x 3
# Groups:   League [4]
  League           data model     
  <chr>  <list<df[,4]>> <list>    
1 F1            [1 x 4] <goalmodl>
2 E0            [1 x 4] <goalmodl>
3 SP1           [1 x 4] <goalmodl>
4 D1            [1 x 4] <goalmodl>

我想使用 goalmodel::predict_result 函数使用其各自的 goalmodl 对象迭代nested_df 上的所有四个数据。提前致谢。

Answer 1

您应该使用map2函数。

map2(.x = nested_df$model,
     .y = nested_df$data,
     .f = ~ predict_result(model = .x,
                           team1 = .y$HomeTeam,
                           team2 = .y$AwayTeam,
                           return_df = TRUE))

您将获得数据帧列表。

或者可能像这样:

nested_df <- nested_df %>% 
  mutate(pred = map2(.x = model,
                     .y = data,
                     .f = ~ predict_result(model = .x,
                                           team1 = .y$HomeTeam,
                                           team2 = .y$AwayTeam,
                                           return_df = TRUE)))

您将在 nested_df 中获得一个列表列。使用 unnest 您可以将其作为数据帧获取。