Wypełnianie kolumny indeksu w data.table

To pytanie jest powiązane z: Dodaj kolumnę do ramki danych, która indeksuje liczbę wystąpień w grupie Mam następującą tabelę danych posortowaną według pierwszych 2 kolumn.

ddt = structure(list(Unit = structure(c(1L, 1L, 2L, 2L, 3L, 3L), .Label = c("A", 
"A1", "B"), class = "factor"), Anything = c(3.4, 6.9, 1.1, 2.2, 
2, 3), index = c(0, 0, 0, 0, 0, 0)), .Names = c("Unit", "Anything", 
"index"), row.names = c(NA, -6L), class = c("data.table", "data.frame"
), .internal.selfref = <pointer: 0x8948f68>, sorted = c("Unit", 
"Anything"))

ddt
   Unit Anything index
1:    A      3.4     0
2:    A      6.9     0
3:   A1      1.1     0
4:   A1      2.2     0
5:    B      2.0     0
6:    B      3.0     0

Kolumnę indeksową należy wypełnić liczbami 1,2,3... dla każdej Jednostki. W przypadku ramki danych mogę to zrobić poprzez:

for(U in unique(ddt$Unit)){
    ddt[ddt$Unit==U,]$index = 1:length(ddt[ddt$Unit==U,]$Unit)
}

ddt
  Unit Anything index
1    A      3.4     1
3    A      6.9     2
4   A1      1.1     1
2   A1      2.2     2
5    B      2.0     1
6    B      3.0     2

Ale jak to zrobić za pomocą poleceń data.table? Dzięki za pomoc.


r data.table
person rnso    schedule 27.08.2014    source źródło

Odpowiedzi (3)


arrow_upward
2
arrow_downward

Próbować

 ddt[, indx:=1:.N, by=Unit]
 #     Unit Anything indx
 #1:    A      3.4    1
 #2:    A      6.9    2
 #3:   A1      1.1    1
 #4:   A1      2.2    2
 #5:    B      2.0    1
 #6:    B      3.0    2
person akrun    schedule 27.08.2014

arrow_upward
1
arrow_downward

Jednym z problemów jest to, że nie można zmienić klasy kolumny za pomocą := (ponieważ indeks jest typu double i idealnie potrzebne są liczby całkowite). Polecam usunąć index i odtworzyć go za pomocą :=:

ddt$index = NULL
ddt[,index:= 1:nrow(.SD), by=Unit]
> ddt
   Unit Anything index
1:    A      3.4     1
2:    A      6.9     2
3:   A1      1.1     1
4:   A1      2.2     2
5:    B      2.0     1
6:    B      3.0     2
person Señor O    schedule 27.08.2014

arrow_upward
1
arrow_downward

Spróbuj tego:

ddt[, index := as.numeric(seq_len(.N)), by="Unit"]
ddt

   Unit Anything index
1:    A      3.4     1
2:    A      6.9     2
3:   A1      1.1     1
4:   A1      2.2     2
5:    B      2.0     1
6:    B      3.0     2
person Andrie    schedule 27.08.2014
comment
użyj .N zamiast tego length urządzenia - person eddi; 27.08.2014
comment
@eddi: Czy możesz wpisać to jako odpowiedź. - person rnso; 27.08.2014