时间戳

首页 > 解决方案 > 如何从 dplyr 数据框中的每个组中选择最后 N 个观察值？

r - 如何从 dplyr 数据框中的每个组中选择最后 N 个观察值？

问题描述

给定一个数据框：

df <- structure(list(a = c(1, 1, 1, 2, 2, 2, 3, 3, 4, 4), b = c(34, 
343, 54, 11, 55, 62, 59, -9, 0, -0.5)), row.names = c(NA, -10L
), class = c("tbl_df", "tbl", "data.frame"))

我想从每组中获取最后 N 个观察值/行：

df %>% 
dplyr::group_by(a) %>% 
dplyr::last(2)

给我错误的结果。

我希望它是：

a   b
1 343
1  54
2  55
2  62
3  59
3  -9
4   0
4  -0.5

请指教这里有什么问题？

我得到的错误是：

顺序错误（order_by）[[n]]：下标超出范围

标签： rdplyr

解决方案

因为这是一个基于的具体问题dplyr

1 ) 在 之后group_by，使用sliceonrow_number()

library(tidyverse)
df %>% 
   group_by(a) %>% 
   slice(tail(row_number(), 2))
# A tibble: 8 x 2
# Groups:   a [4]
#      a      b
#  <dbl>  <dbl>
#1     1  343  
#2     1   54  
#3     2   55  
#4     2   62  
#5     3   59  
#6     3   -9  
#7     4    0  
#8     4   -0.5

2 ) 或使用filterfromdplyr

df %>% 
   group_by(a) %>% 
   filter(row_number() >= (n() - 1))

3 ) 或与do和tail

df %>%
    group_by(a) %>% 
    do(tail(., 2))

4 ) 除了tidyverse, 方法，我们还可以使用 compactdata.table

library(data.table)
setDT(df)[df[, .I[tail(seq_len(.N), 2)], a]$V1]

5 ) 或by从base R

by(df, df$a, FUN = tail, 2)

6 ) 或aggregate从base R

df[aggregate(c ~ a, transform(df, c = seq_len(nrow(df))), FUN = tail, 2)$c,]

7 ) 或split从base R

do.call(rbind, lapply(split(df, df$a), tail, 2))

推荐阅读