分组汇总仍然给出每个单独行的结果

Question

我有以下数据：

library(tidyverse)

df <- data.frame(id = c(1,1,1,2,2,2),
                 x = rep(letters[1:2], each = 3),
                 y = c(3,4,3,5,6,5),
                 z = c(7,8,9,10,11,12))

我现在想按 id 汇总数据，根据 y 值获取 z 总和。 y 条件本身取决于 x 的值。

我以为我可以使用下面的代码，但这给了我所有输入 ID 并且没有总结。结果是正确的，但我仍然希望每个 id 一行。

df %>%
  group_by(id) %>%
  summarize(test = case_when(x == 'a' ~ sum(z[y == 3]),
                             x == 'b' ~ sum(z[y == 5])))

# A tibble: 6 x 2
# Groups:   id [2]
     id  test
  <dbl> <dbl>
1     1    16
2     1    16
3     1    16
4     2    22
5     2    22
6     2    22

以下有效，但我不明白为什么它有效，而上面的代码却无效。

df %>%
  group_by(id) %>%
  summarize(test = case_when(all(x == 'a') ~ sum(z[y == 3]),
                             all(x == 'b') ~ sum(z[y == 5])))

# A tibble: 2 x 2
     id  test
  <dbl> <dbl>
1     1    16
2     2    22

另外，有没有更直接的方法来进行总结？

Answer 1

因为，与 ifelse(test, x, y) 类似的 case_when 会返回与 test 长度相同的向量。 all(x == z) 的长度为 1，因此返回值的长度为 1。