Question

在下面的数据表中，我想选择具有唯一id且行pos2中的最小值的行：

dt = data.table(id = c(1,2,2,3,3,3),
            pos1 = c(0.1, 0.2, 0.2, 0.3, 0.3, 0.3),
            pos2 = c(0.1, 0.25, 0.21, 0.34, 0.31, 0.32))

   id pos1 pos2
1:  1  0.1 0.10
2:  2  0.2 0.25
3:  2  0.2 0.21
4:  3  0.3 0.34
5:  3  0.3 0.31
6:  3  0.3 0.32

我现在要做的是创建一个中间表：

dt.red = dt[, .(pos2 = first(sort(pos2))), by = id]

   id pos2
1:  1 0.10
2:  2 0.21
3:  3 0.31

然后我合并以获得所需的最终结果：

merge(dt, dt.red)

   id pos2 pos1
1:  1 0.10  0.1
2:  2 0.21  0.2
3:  3 0.31  0.3

是否有更干净的方法可以通过data.table实现这一目标？

Answer 1

也可以在没有.I的情况下执行此操作，但速度会较慢*

dt[order(pos2), head(.SD, 1), id]
#    id pos1 pos2
# 1:  1  0.1 0.10
# 2:  2  0.2 0.21
# 3:  3  0.3 0.31

*也许不是，请参见下面的评论

Answer 2

我们可以使用.I来获取行索引，并在i中使用它来对行进行子集

dt[dt[order(pos2), .I[1], by = id]$V1]
#   id pos1 pos2
#1:  1  0.1 0.10
#2:  2  0.2 0.21
#3:  3  0.3 0.31

或使用setorder

setorder(dt, id, pos2)[, .SD[1L], id]

data.table：按索引的子集

2 个答案: