Как выбрать N-й элемент в подгруппе с помощью data.table?

Простая задача: мне нужно извлечь (скажем) 1-й и 2-й элементы подгруппы.

DT <- data.table (X=c(1,2), Y=round(rnorm(5),1), key = "X"); DT

       X     Y
1:     1  -0.4
2:     1   0.5
3:     1   1.2
4:     2   1.0
5:     2  -0.3

Итак, мне нужно получить DT2, который будет

1:     1  -0.4
2:     1   0.5
3:     2   1.0
4:     2  -0.3

Это не работает:

 DT2 <- DT[, .(first = .SD[1], second=.SD[2]), by=X]

Что работает?

.SD означает подмножество data.table, набор столбцов. То, что вы хотели присвоить first и second, — это отдельные столбцы. Просто используйте Y[1L] и Y[2L] вместо .SD.

— 21.04.2019 16:55

r data.table

18.04.2019 20:44

Стоит ли изучать PHP в 2023-2024 годах?

Привет всем, сегодня я хочу высказать свои соображения по поводу вопроса, который я уже много раз получал в своем сообществе: "Стоит ли изучать PHP в...

Поведение ключевого слова "this" в стрелочной функции в сравнении с нормальной функцией

В JavaScript одним из самых запутанных понятий является поведение ключевого слова "this" в стрелочной и обычной функциях.

Приемы CSS-макетирования - floats и Flexbox

Здравствуйте, друзья-студенты! Готовы совершенствовать свои навыки веб-дизайна? Сегодня в нашем путешествии мы рассмотрим приемы CSS-верстки - в...

Тестирование функциональных ngrx-эффектов в Angular 16 с помощью Jest

В системе управления состояниями ngrx, совместимой с Angular 16, появились функциональные эффекты. Это здорово и делает код определенно легче для...

Концепция локализации и ее применение в приложениях React ⚡️

Локализация - это процесс адаптации приложения к различным языкам и культурным требованиям. Это позволяет пользователям получить опыт, соответствующий...

Пользовательский скаляр GraphQL

Листовые узлы системы типов GraphQL называются скалярами. Достигнув скалярного типа, невозможно спуститься дальше по иерархии типов. Скалярный тип...

193

Перейти к ответу Данный вопрос помечен как решенный

Ответы 4

Можем создать группу head

DT[,  head(.SD, 2), X]

Или извлеките индекс строки (.I) и подмножество

DT[DT[, .I[1:2], X]$V1]

18.04.2019 20:46

Вот решение с использованием tidyverse/dplyr. Первый раздел использует tibble, который является расширением обычного фрейма данных.

library(tidyverse)

df <- tibble(
  x = c(1, 1, 1, 2, 2),
  y = c(-0.4, 0.5, 1.2, 1.0, -0.3)
)

df
# A tibble: 5 x 2
      x     y
  <dbl> <dbl>
1     1  -0.4
2     1   0.5
3     1   1.2
4     2   1  
5     2  -0.3

df2 <-
  df %>% 
  group_by(x) %>% 
  slice(1:2) %>% 
  ungroup()

df2
# A tibble: 4 x 2
      x     y
  <dbl> <dbl>
1     1  -0.4
2     1   0.5
3     2   1  
4     2  -0.3

18.04.2019 21:48

Следующий код также дает тот же результат. Если порядок важен, setkey или order можно запустить заранее.

DT <- data.table (X=c(1,2), Y=round(rnorm(5),1), key = "X"); DT
   X    Y
1: 1  0.5
2: 1  1.4
3: 1  0.0
4: 2 -1.5
5: 2 -0.3

DT[,Y[1:2],.(X)]

   X   V1
1: 1  0.5
2: 1  1.4
3: 2 -1.5
4: 2 -0.3

19.04.2019 19:18

Ответ принят как подходящий

Вот как я бы это сделал, и я думаю, что способ предназначена (авторы пакета):