Wybierz pierwszy i ostatni wiersz z danych zgrupowanych

Question

Pytanie

Korzystanie dplyr, jak mam wybrać górne i dolne obserwacje/wiersze danych zgrupowanych w jednym stwierdzeniem?

danych & Przykład

względu ramka danych

df <- data.frame(id=c(1,1,1,2,2,2,3,3,3), 
       stopId=c("a","b","c","a","b","c","a","b","c"), 
       stopSequence=c(1,2,3,3,1,4,3,1,2)) 

można uzyskać górną i obserwacje dolnej z każdej z grup, stosując slice, ale za pomocą dwóch oddzielnych statments:

firstStop <- df %>% 
    group_by(id) %>% 
    arrange(stopSequence) %>% 
    slice(1) %>% 
    ungroup 

lastStop <- df %>% 
    group_by(id) %>% 
    arrange(stopSequence) %>% 
    slice(n()) %>% 
    ungroup 

Czy mogę połączyć te dwa statmenety w jeden, który wybiera zarówno obserwacje górne i dolne?

Answer 1

Jest prawdopodobnie szybszy sposób:

df %>% 
    group_by(id) %>% 
    arrange(stopSequence) %>% 
    filter(row_number()==1 | row_number()==n()) 

Answer 2

Coś jak:

library(dplyr) 

df <- data.frame(id=c(1,1,1,2,2,2,3,3,3), 
       stopId=c("a","b","c","a","b","c","a","b","c"), 
       stopSequence=c(1,2,3,3,1,4,3,1,2)) 

first_last <- function(x) { 
    bind_rows(slice(x, 1), slice(x, n())) 
} 

df %>% 
    group_by(id) %>% 
    arrange(stopSequence) %>% 
    do(first_last(.)) %>% 
    ungroup 

## Source: local data frame [6 x 3] 
## 
## id stopId stopSequence 
## 1 1  a   1 
## 2 1  c   3 
## 3 2  b   1 
## 4 2  c   4 
## 5 3  b   1 
## 6 3  a   3 

Z do można bardzo dużo wykonać dowolną liczbę operacji na grupy, ale na @ jeremycg odpowiedź jest bardziej odpowiednie tylko dla tego zadania.

Answer 3

Nie dplyr, ale jest to o wiele bardziej bezpośredni za pomocą data.table:

library(data.table) 
setDT(df) 
df[ df[order(id, stopSequence), .I[c(1L,.N)], by=id]$V1 ] 
# id stopId stopSequence 
# 1: 1  a   1 
# 2: 1  c   3 
# 3: 2  b   1 
# 4: 2  c   4 
# 5: 3  b   1 
# 6: 3  a   3 

Bardziej szczegółowe wyjaśnienie:

# 1) get row numbers of first/last observations from each group 
# * basically, we sort the table by id/stopSequence, then, 
#  grouping by id, name the row numbers of the first/last 
#  observations for each id; since this operation produces 
#  a data.table 
# * .I is data.table shorthand for the row number 
# * here, to be maximally explicit, I've named the variable V1 
#  as row_num to give other readers of my code a clearer 
#  understanding of what operation is producing what variable 
first_last = df[order(id, stopSequence), .(row_num = .I[c(1L,.N)]), by=id] 
idx = first_last$row_num 

# 2) extract rows by number 
df[idx] 

Koniecznie sprawdź Getting Started wiki dla coraz data.table podstawy pokryty

Answer 4

Tylko dla kompletności: Możesz podać slice wektor indeksu s:

df %>% arrange(stopSequence) %>% group_by(id) %>% slice(c(1,n())) 

co daje

id stopId stopSequence 
1 1  a   1 
2 1  c   3 
3 2  b   1 
4 2  c   4 
5 3  b   1 
6 3  a   3 

Answer 5

Znam pytanie określony dplyr. Ale ponieważ inni już napisali rozwiązań korzystania z innych pakietów, postanowiłem mieć go przy użyciu innych pakietów zbyt:

pakiet podstawowy:

df <- df[with(df, order(id, stopSequence, stopId)), ] 
merge(df[!duplicated(df$id), ], 
     df[!duplicated(df$id, fromLast = TRUE), ], 
     all = TRUE) 

danych.Tabela:

df <- setDT(df) 
df[order(id, stopSequence)][, .SD[c(1,.N)], by=id] 

sqldf:

library(sqldf) 
min <- sqldf("SELECT id, stopId, min(stopSequence) AS StopSequence 
     FROM df GROUP BY id 
     ORDER BY id, StopSequence, stopId") 
max <- sqldf("SELECT id, stopId, max(stopSequence) AS StopSequence 
     FROM df GROUP BY id 
     ORDER BY id, StopSequence, stopId") 
sqldf("SELECT * FROM min 
     UNION 
     SELECT * FROM max") 

W jednym zapytania:

sqldf("SELECT * 
     FROM (SELECT id, stopId, min(stopSequence) AS StopSequence 
       FROM df GROUP BY id 
       ORDER BY id, StopSequence, stopId) 
     UNION 
     SELECT * 
     FROM (SELECT id, stopId, max(stopSequence) AS StopSequence 
       FROM df GROUP BY id 
       ORDER BY id, StopSequence, stopId)") 

wyjściowa:

id stopId StopSequence 
1 1  a   1 
2 1  c   3 
3 2  b   1 
4 2  c   4 
5 3  a   3 
6 3  b   1