使用 do 时保留列

Question

代码

假设我有以下代码（我知道，而不是第二个do，mutate在这种情况下我可以使用一个简单的（并跳过rowwise()），但这不是重点，因为在我的实际代码中，第二do个更复杂一些并且计算一个模型）：

library(dplyr)
set.seed(1)
d <- data_frame(n = c(5, 1, 3))
e <- d %>% group_by(n) %>% 
    do(data_frame(y = rnorm(.$n), dat = list(data.frame(a = 1)))) 
e %>% rowwise() %>% do(data_frame(sum = .$y + .$n))

# Source: local data frame [9 x 1]
# Groups: <by row>

# # A tibble: 9 x 1
#         sum
# *     <dbl>
# 1 0.3735462
# 2 3.1836433
# 3 2.1643714
# 4 4.5952808
# 5 5.3295078
# 6 4.1795316
# 7 5.4874291
# 8 5.7383247
# 9 5.5757814

问题

如您所见，结果仅包含列sum。

题

有没有一种方法，以保持原来的列从e 而不需要显式地指定它们（如在e %>% do(data_frame(n = .$n, y = .$y, dat = .$dat, sum = .$y + .$n))在dplyr或做我必须使用purrrlyr::by_row？（不，我不喜欢purrrlyr*，我只是想知道是否有直接的dplyr做这件事的方式我可能忽略了）：

e %>% purrrlyr::by_row(function(x) x$y + x$n, .collate = "cols", .to = "sum")

---

*) 好吧，实际上有一个问题purrrlyr::by_row：

e %>% purrrlyr::by_row(function(x) data_frame(sum = x$y + x$n, diff = x$y - x$n), 
                       .collate ="cols")

将产生列sum1和diff1我需要再次重命名为得到sum和diff，这增加了另一行代码。

Answer 1

我几乎从不使用do，而是将nest,mutate和结合起来使用map。

很难说在你的情况下会是什么样子，因为你的例子似乎没有完全说明你的需求。

在最简单的情况下，您可以指定您确实需要的变量（例如，如果它们是 S3 对象列表）：

mutate(e, sum = map2_dbl(y, n, `+`))

或者，您可以嵌套所需的数据，然后映射整个数据。例如：

f <- e
f$r <- 1:nrow(e) # i.e. add some other variable, not necessarily row indices

f %>%
  ungroup() %>%                               # e was still grouped
  nest(n:dat) %>%                               # specify what you variables you need
  mutate(sum = map_dbl(data, ~.$y + .$n)) %>% # map to data, use the same formula as in do
  unnest()                                    # unnest to get original columns back

两者都保留原始列不变。

对于建模示例，例如：

mtcars %>% 
  group_by(cyl) %>% 
  nest() %>% 
  mutate(model = map(data, ~lm(qsec ~ hp, .)),
         coef  = map_dbl(model, ~coef(.)[2])) %>% 
  unnest(data)

这将为您提供所有原始数据，但每组增加了回归系数。在取消嵌套之前，整个模型都在您的 data.frame 中作为列表列。