sacar - Cuente el número de registros y genere el número de fila dentro de cada grupo en una tabla de datos.

que es un periodo en la tabla periodica

Question

Tengo los siguientes datos.

set.seed(1)
DT <- data.table(VAL = sample(c(1, 2, 3), 10, replace = TRUE))
VAL
1: 1
2: 2
3: 2
4: 3
5: 1
6: 3
7: 3
8: 2
9: 2
10: 1

Dentro de cada número en VAL quiero:

1. Cuente el número de registros / filas
2. Cree un índice de fila (contador) de primera, segunda, tercera aparición, etc.

Al final quiero el resultado.

 VAL COUNT IDX
1: 1 3 1
2: 2 4 1
3: 2 4 2
4: 3 3 1
5: 1 3 2
6: 3 3 2
7: 3 3 3
8: 2 4 3
9: 2 4 4
10: 1 3 3

donde "COUNT" es el número de registros / filas para cada "VAL", e "IDX" es el índice de fila dentro de cada "VAL".

Intenté trabajar con which y length usando .I :

 dt[, list(COUNT = length(VAL == VAL[.I]),
IDX = which(which(VAL == VAL[.I]) == .I))]

pero esto no funciona como .I refiere a un vector con el índice, así que supongo que uno debe usar .I[] . Aunque dentro de .I[] otra vez enfrento el problema, no tengo el índice de filas y sí sé (por leer las data.table frecuentes de la data.table y seguir las publicaciones aquí) que se debe evitar el bucle a través de las filas si es posible.

Entonces, ¿cuál es la forma data.table ?

Answer 1

Utilizando .N ...

DT[ , `:=`( COUNT = .N , IDX = 1:.N ) , by = VAL ]
# VAL COUNT IDX
# 1: 1 3 1
# 2: 2 4 1
# 3: 2 4 2
# 4: 3 3 1
# 5: 1 3 2
# 6: 3 3 2
# 7: 3 3 3
# 8: 2 4 3
# 9: 2 4 4
#10: 1 3 3

.N es el número de registros en cada grupo, con grupos definidos por "VAL" .