R数据表“by”用于“i”

Question

在给定的数据表中，如何选择V3分组的V2 = max(V2)的列V1的语法。

例如： 在mtcars数据集中，我想找出与max(disp)相等的观测值相对应的hp，按cyl分组。

以下是我的不优雅的解决方案，使用which：

mtcars <- data.table(mtcars)
mtcars[which(mtcars$disp %in% mtcars[, max(disp), by = .(cyl)]$V1), .(cyl,hp)]
   cyl  hp
1:   6 110
2:   4  62
3:   8 205

有没有更多“data.table”方式实现相同的结果？

- Ivan P.

---

还可以在这里看一下：https://dev59.com/1WQn5IYBdhLWcg3wxZni - talat

Answer 1

我们可以尝试使用“join”函数。

mtcars[mtcars[, list(disp=max(disp)), by = cyl], 
    on = c('cyl', 'disp')][, c('cyl', 'hp'), with=FALSE]
#   cyl  hp
#1:   6 110
#2:   4  62
#3:   8 205

---

这里有一个更简短的方法可以得到期望的输出。

mtcars[, .SD[disp==max(disp), .(hp)], by = cyl]
#   cyl  hp
#1:   6 110
#2:   4  62
#3:   8 205

- akrun

Answer 2

您可以使用.I：

mtcars[mtcars[, .I[which.max(disp)], by = cyl]$V1, .(cyl, hp)]
#   cyl  hp
#1:   6 110
#2:   4  62
#3:   8 205

或者

mtcars[, hp[disp == max(disp)], by=cyl]
#   cyl  V1
#1:   6 110
#2:   4  62
#3:   8 205

你可以稍微简化自己的方法：

mtcars[disp %in% mtcars[, max(disp), by = .(cyl)]$V1, .(cyl,hp)]

- talat

Answer 3

mtcars[,ismax:=disp==max(disp),by=cyl][ismax==T, .(cyl, hp)]
   cyl  hp
1:   6 110
2:   4  62
3:   8 205

我总是得到和其他人几乎完全相同的解决方案。哎，算了...

mtcars[,.SD[disp == max(disp)], by = cyl, .SDcols = c("disp", "hp")][,.(cyl, hp)]

- statespace

---

该方法在原始数据中不必要地创建了新列。 - akrun

好的，这个确实看起来是一个简短而容易的解决方案。让我想一想如何在不创建新列的情况下产生相同的输出。 - statespace