Seleccione la primera y la última fila de los datos agrupados


137

Pregunta

Utilizando dplyr, ¿cómo selecciono las observaciones / filas superiores e inferiores de datos agrupados en una declaración?

Datos y ejemplo

Dado un marco de datos

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))

Puedo obtener las observaciones superior e inferior de cada grupo usando slice, pero usando dos declaraciones separadas:

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

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

¿Puedo combinar estos dos statmenets en uno que seleccione las observaciones superiores e inferiores?


Respuestas:


232

Probablemente hay una forma más rápida:

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

66
rownumber() %in% c(1, n())evitaría la necesidad de ejecutar el escaneo vectorial dos veces
MichaelChirico

13
@MichaelChirico Sospecho que omitiste un _? es decirfilter(row_number() %in% c(1, n()))
Eric Fail

107

Solo para completar: puede pasar sliceun vector de índices:

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

lo que da

  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

incluso podría ser más rápido que filter- no he probado esto, pero mira aquí
Tjebo

1
@Tjebo A diferencia del filtro, el segmento puede devolver la misma fila varias veces, por ejemplo, mtcars[1, ] %>% slice(c(1, n()))en ese sentido, la elección entre ellos depende de lo que desea que se devuelva. Esperaría que los tiempos sean cercanos a menos que nsea ​​muy grande (donde se podría favorecer el corte), pero tampoco lo he probado.
Frank

15

No dplyr, pero es mucho más directo usando 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

Explicación más detallada:

# 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]

Asegúrese de revisar la Getting Started wiki para conseguir los data.tablefundamentos cubiertos


1
O df[ df[order(stopSequence), .I[c(1,.N)], keyby=id]$V1 ]. Ver idaparecer dos veces es extraño para mí.
Frank

Puede configurar claves en la setDTllamada. Entonces una orderllamada no es necesaria aquí.
Artem Klevtsov

1
@ArtemKlevtsov: sin embargo, es posible que no siempre desee configurar las teclas.
SymbolixAU

2
O df[order(stopSequence), .SD[c(1L,.N)], by = id]. Ver aquí
JWilliman

@JWilliman que no necesariamente será exactamente el mismo, ya que no se volverá a ordenar id. Creo que df[order(stopSequence), .SD[c(1L, .N)], keyby = id]debe hacer el truco (con la diferencia menor a la solución anterior que el resultado será keyed
MichaelChirico

8

Algo como:

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

Con dousted puede realizar cualquier cantidad de operaciones en el grupo, pero la respuesta de @ jeremycg es mucho más apropiada para esta tarea.


1
No había considerado escribir una función, ciertamente una buena forma de hacer algo más complejo.
tospig

1
Esto parece demasiado complicado en comparación con solo usar slice, comodf %>% arrange(stopSequence) %>% group_by(id) %>% slice(c(1,n()))
Frank

44
No estoy en desacuerdo (y señalé que jeremycg es una mejor respuesta en la publicación), pero tener un doejemplo aquí podría ayudar a otros cuando sliceno funcionen (es decir, operaciones más complejas en un grupo). Y, debe publicar su comentario como respuesta (es el mejor).
hrbrmstr

6

Sé la pregunta especificada dplyr . Pero, dado que otros ya publicaron soluciones con otros paquetes, decidí probar también con otros paquetes:

Paquete base:

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

tabla de datos:

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")

En una consulta:

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)")

Salida:

  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

3

usando which.miny which.max:

library(dplyr, warn.conflicts = F)
df %>% 
  group_by(id) %>% 
  slice(c(which.min(stopSequence), which.max(stopSequence)))

#> # A tibble: 6 x 3
#> # Groups:   id [3]
#>      id stopId stopSequence
#>   <dbl> <fct>         <dbl>
#> 1     1 a                 1
#> 2     1 c                 3
#> 3     2 b                 1
#> 4     2 c                 4
#> 5     3 b                 1
#> 6     3 a                 3

punto de referencia

También es mucho más rápido que la respuesta actual aceptada porque encontramos el valor mínimo y máximo por grupo, en lugar de ordenar toda la columna stopSequence.

# create a 100k times longer data frame
df2 <- bind_rows(replicate(1e5, df, F)) 
bench::mark(
  mm =df2 %>% 
    group_by(id) %>% 
    slice(c(which.min(stopSequence), which.max(stopSequence))),
  jeremy = df2 %>%
    group_by(id) %>%
    arrange(stopSequence) %>%
    filter(row_number()==1 | row_number()==n()))
#> Warning: Some expressions had a GC in every iteration; so filtering is disabled.
#> # A tibble: 2 x 6
#>   expression      min   median `itr/sec` mem_alloc `gc/sec`
#>   <bch:expr> <bch:tm> <bch:tm>     <dbl> <bch:byt>    <dbl>
#> 1 mm           22.6ms     27ms     34.9     14.2MB     21.3
#> 2 jeremy      254.3ms    273ms      3.66    58.4MB     11.0

2

Utilizando data.table:

# convert to data.table
setDT(df) 
# order, group, filter
df[order(stopSequence)][, .SD[c(1, .N)], by = id]

   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

1

Otro enfoque con lapply y una declaración dplyr. Podemos aplicar un número arbitrario de cualquier función de resumen a la misma declaración:

lapply(c(first, last), 
       function(x) df %>% group_by(id) %>% summarize_all(funs(x))) %>% 
bind_rows()

Por ejemplo, podría estar interesado en filas con el valor max stopSequence y hacer:

lapply(c(first, last, max("stopSequence")), 
       function(x) df %>% group_by(id) %>% summarize_all(funs(x))) %>%
bind_rows()

0

Una alternativa diferente base de R sería primero orderpor idy stopSequence, splitellos basados en idy para cada idseleccionamos sólo el primero y el último índice y el subconjunto de la trama de datos usando esos índices.

df[sapply(with(df, split(order(id, stopSequence), id)), function(x) 
                   c(x[1], x[length(x)])), ]


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

O similar usando by

df[unlist(with(df, by(order(id, stopSequence), id, function(x) 
                   c(x[1], x[length(x)])))), ]
Al usar nuestro sitio, usted reconoce que ha leído y comprende nuestra Política de Cookies y Política de Privacidad.
Licensed under cc by-sa 3.0 with attribution required.