用dplyr折叠行

时间:2017-10-28 21:56:10

标签: r group-by dplyr summarize

我是R的新手,我正在尝试使用dplyr基于行值折叠行。以下示例显示了示例数据。

set.seed(123)

df<-data.frame(A=c(rep(1:4,4)),
               B=runif(16,min=0,max=1),
               C=rnorm(16, mean=1,sd=0.5))

   A B          c
1  1 0.36647435 0.7485365
2  2 0.51864614 0.8654337
3  3 0.04596929 0.9858012
4  4 0.15479619 1.1294208
5  1 0.76712372 1.2460700
6  2 0.17666676 0.7402996
7  3 0.89759874 1.2699954
8  4 0.90267735 0.7101804
9  1 0.91744223 0.3451281
10 2 0.25472599 0.8604743
11 3 0.10933985 0.8696796
12 4 0.71656017 1.2648846
13 1 0.21157810 1.3170205
14 2 0.14947268 1.2789700
15 3 0.92251060 1.5696901
16 4 0.30090579 1.7642853

我想根据条件A中的行将值1和2作为一行(作为第1行和第2行的平均值)来汇总/折叠两行。因此,最终结果将只有12行,因为其他4行已折叠。

我尝试使用以下dplyr功能,但收效甚微。

install.packages(“tidyverse”) 图书馆(tidyverse)

df %>% summarize_each( fun(i){ for i %in% c(1,2)funs(mean) })

预期输出类似于:

   A    B           C
1  1.5  0.4425602   0.8069851
3  3    0.04596929  0.9858012
4  4    0.15479619  1.1294208
5  1.5  0.4718952   0.9931848
7  3    0.89759874  1.2699954
8  4    0.90267735  0.7101804
9  1.5  0.5860841   0.6028012
11 3    0.10933985  0.8696796
12 4    0.71656017  1.2648846
13 1.5  0.1805254   1.297995
15 3    0.92251060  1.5696901
16 4    0.30090579  1.7642853

提前谢谢。

2 个答案:

答案 0 :(得分:1)

一种选择是将.language-select{ background-color:#fff; padding: 5px 5px; width: 20px; } 等于1或2的行与其他行分开处理,然后将它们重新绑定在一起:

A
set.seed(3)
df<-data.frame(A=c(rep(1:4,4)),B=runif(16,min=0,max=1),c=rnorm(16, mean=1,sd=0.5))

df %>% 
  filter(A %in% 1:2) %>% 
  group_by(tmp=cumsum(A==1)) %>% 
  summarise_all(mean) %>% 
  ungroup %>% select(-tmp) %>% 
  bind_rows(df %>% filter(!A %in% 1:2))

答案 1 :(得分:1)

通过明确隐式的基于顺序的分组,摘要可以 通过一次summarise_all电话完成。

# Generate the data
set.seed(1)

df <- data.frame(
  A = c(rep(1:4, 4)), 
  B = runif(16, min = 0, max = 1), 
  C = rnorm(16, mean = 1, sd = 0.5)
)

library(dplyr)

new <- df %>%
  group_by(grp = rep(
    1:4,      # vector containing names of groups to create
    each = 4  # number of elements in each group
  )) %>% 
  group_by(mean_grp = cumsum(A > 2) + 1, add = T) %>%
  summarise_all(mean) %>%
  ungroup()

new
#> # A tibble: 12 x 5
#>      grp mean_grp     A         B           C
#>    <int>    <dbl> <dbl>     <dbl>       <dbl>
#>  1     1        1   1.5 0.3188163 1.067598241
#>  2     1        2   3.0 0.5728534 1.755890584
#>  3     1        3   4.0 0.9082078 1.194921618
#>  4     2        1   1.5 0.5500358 0.291014883
#>  5     2        2   3.0 0.9446753 1.562465459
#>  6     2        3   4.0 0.6607978 0.977533195
#>  7     3        1   1.5 0.3454502 1.231911487
#>  8     3        2   3.0 0.2059746 1.410610598
#>  9     3        3   4.0 0.1765568 1.296950661
#> 10     4        1   1.5 0.5355633 1.425278418
#> 11     4        2   3.0 0.7698414 1.037282492
#> 12     4        3   4.0 0.4976992 0.005324152

我建议在分组之后将分组变量保留在数据中 摘要(如果首先包含它们,一切都会更简单), 但如果你愿意的话,你可以放弃它们 new %>% select(-grp, -mean_grp)

PS。为了避免在代码中包含"magic numbers"(例如创建1:4each = 4grp),您还可以创建第一个分组变量:

grp = cumsum(A < lag(A, default = A[1])) + 1

假设订购原始数据,以便每当A的值小于之前的A值时,就会启动新的组。