Фильтр data.table по количеству группы

Question

Допустим, у меня есть data.table как

sample<-data.table(id=c(1,1,2,2,3,3,3,4,4), 
        name=c("apple","apple","orange","orange", 
          "pear","pear","pear","banana","banana"), 
       atr=c("pretty","ugly","bruised","delicious", 
        "pear-shaped","bruised","infested", 
        "too-ripe","perfect"), 
       N=c(10,9,15,4,5,7,7,4,12)) 

Я хочу вернуться в основном unique(sample[,list(id, name)]) кроме того, что я также хочу atr столбец значения с наибольшим N. В тех случаях, когда есть связь для самого высокого N, тогда мне все равно, какая из двух выбрана, но я хочу, чтобы только один был выбран.

Это почти работает merge(sample[,list(N=max(N)),by=list(id,name1)], sample,by=c("id","name1","N")), но поскольку груша имеет два значения, которые связывают для максимального, это возвращает две груши. Кроме того, что он не дает ожидаемого результата, я также предполагаю/надеюсь, что есть способ сделать это, не связанный с объединением.

Answer 1

Я бы просто использование order:

> unique(sample[order(-N), .(id, name, atr)], by = c("id", "name")) 
    id name  atr 
1: 2 orange bruised 
2: 4 banana perfect 
3: 1 apple pretty 
4: 3 pear bruised 

Если вы хотите m aintain общей сортировки, просто используйте вместо этого order(id, name, -N).

Вы также можете разбить это на две линии:

setorder(sample, -N) #done by reference, as with all set* functions in data.table 
unique(sample[ , .(id, name, atr)], by = c("id", "name")) 

или, возможно, лучше, в зависимости от вашей конечной цели:

setkey(setorder(sample, -N), id, name) 
unique(sample[ , .(id, name, atr)]) 

(примечание: порядок имеет решающее значение в этой последней, как использование setorder первый перезаписывает ключи до NULL)

Answer 2

Вы можете использовать atr[N == max(N)][1] вернуть только первый в случае равенства, как это -

library(data.table) 

sample[, .(atr = atr[N == max(N)][1]), by = .(id, name)] 
# id name  atr 
# 1: 1 apple pretty 
# 2: 2 orange bruised 
# 3: 3 pear bruised 
# 4: 4 banana perfect 

Примечание: Как Франк указывает atr[N == max(N)][1] также просто atr[which.max(N)]