使用 dplyr 返回函数列表

Question

我有一个带有 x,y 坐标和分组变量的 data.frame：

df<-data.frame(group=c(rep(1,3),rep(2,3),rep(3,3)),
               x=c(rep(c(1,2,3),3)),
               y=c(1,2,3,1,4,9,1,8,27))

我想为每个组安装一个样条函数以供以后使用。类似于：

df %>% group_by(group) %>% .... foo(splinefun(x,y))

我不关心原始数据（其中包含多个组、x、y），因此返回值可以是函数的命名列表，或者我可以使用组变量引用并获得拟合样条函数的任何内容.用例是迭代具有组和插值坐标 x 的第二个数据集。 有人吗？

编辑

因此，更新后的示例 newData 包含一个未包含在拟合函数标题中的组：

df<-data.frame(group=c(rep(1,4),rep(2,4),rep(3,4)),
               x=c(rep(c(0,1,2,3),3)),
               y=c(0,1,2,3,0,1,4,9,0,1,8,27))

fit<-df %>% 
  group_by(group) %>% 
  summarise(fns = list(approxfun(x, y)))

newData <- data.frame(group=c(rep(1,5),rep(2,5),rep(3,5),rep(4,5)) , 
                      xval = rep(c(0,1,2,3,4),4)) %>% 
                      left_join(fit)

因此，拟合的幂函数现在与扩展的组集相连接，其中列 fns 在缺失记录上为 NULL。如果我现在可以使用 interpolated 列应用 fns 不为 NULL 且不为 NA 的函数来改变 newData，否则我就完成了。

Answer 1

我们可以用list包装输出函数

library(tidyverse)
df %>% 
    group_by(group) %>%
    summarise(fns = list(splinefun(x, y)))

如果打算在原始数据中创建一列，请将summarise 替换为mutate。但是，这将对组的每一行重复该功能

---

如果我们想在相应“组”上的另一个数据集上应用函数，则与第二个数据集进行连接，并使用map2 将存储在第一个数据列中的函数应用到第二个数据的感兴趣列上

df %>% 
   group_by(group) %>% 
   summarise(fns = list(splinefun(x, y))) %>% 
   left_join(df2) %>% 
   mutate(newval = map2_dbl(fns, xval, ~ .x(.y)))

更新

基于 OP 帖子中的更新数据

fit %>% 
  right_join(newData) %>% 
  mutate(newval = map2_dbl(fns, xval, ~ if(!is.null(.x)) .x(.y) else NA_real_))

或者使用possibly（来自purrr）创建一个函数并应用

f1 <- possibly(function(x, y) x(y), otherwise = NA_real_)
fit %>% 
  right_join(newData) %>% 
  mutate(newval = map2_dbl(fns, xval, f1))

数据

df2 <- data.frame(group = 1:3, xval = c(1.5, 2.5, 3.2))
newData <- data.frame(group=c(rep(1,5),rep(2,5),rep(3,5),rep(4,5)), 
                   xval = rep(c(0,1,2,3,4),4))