Как подсчитать комбинации в таблице данных в R

У меня есть два data.tables. Я хотел бы подсчитать количество строк, соответствующих комбинации таблицы в другой таблице. Я проверил документацию data.table, но я не нашел ответа. Я использую data.table 1.9.2.

DT1 <- data.table(a=c(3,2), b=c(8,3))
DT2 <- data.table(w=c(3,3,3,2,3), x=c(8,8,8,3,7), z=c(2,6,7,2,2))
DT1
#    a b
# 1: 3 8
# 2: 2 3

DT2
#    w x z
# 1: 3 8 2
# 2: 3 8 6
# 3: 3 8 7
# 4: 2 3 2
# 5: 3 7 2

Теперь я хотел бы подсчитать количество пар (3, 8) и (2, 3) пар в DT2.

setkey(DT2, w, x)
nrow(DT2[J(3, 8), nomatch=0])
# [1] 3    ## OK !

nrow(DT2[J(2, 3), nomatch=0])
# [1] 1    ## OK !

DT1[,count_combination_in_dt2 := nrow(DT2[J(a, b), nomatch=0])]
DT1
#    a b count_combination_in_dt2
# 1: 3 8                        4 ## not ok.
# 2: 2 3                        4 ## not ok.

Ожидаемый результат:

#    a b count_combination_in_dt2
# 1: 3 8                        3 
# 2: 2 3                        1

Ответы

Ответ 1

Вам просто нужно добавить by=list(a,b).

DT1[,count_combination_in_dt2:=nrow(DT2[J(a,b),nomatch=0]), by=list(a,b)]
DT1
## 
##    a b count_combination_in_dt2
## 1: 3 8                        3
## 2: 2 3                        1

EDIT: Еще несколько деталей. В исходной версии вы использовали DT2[DT1, nomatch=0] (потому что вы использовали все комбинации a, b. Если вы хотите использовать J(a,b) для каждой комбинации a, b отдельно, вам нужно использовать by. data.table затем сгруппирован по a, b, а nrow(...) оценивается внутри каждой группы.

Ответ 2

setkey(DT2, w, x)

DT2[DT1, .N, by = .EACHI]
#   w x N
#1: 3 8 3
#2: 2 3 1

# In versions <= 1.9.2, use DT2[DT1, .N] instead

Выше просто выполняется слияние и подсчитывается количество строк для каждой группы, определенных i-expression, таким образом by = .EACHI.