计算每组 data.table 窗口中值的数量答案

【问题标题】：Count the number of values in a window of a data.table per group计算每组 data.table 窗口中值的数量
【发布时间】：2016-03-14 16:42:06
【问题描述】：

我正在尝试向data.table 添加新列，其中行中的值取决于行中值的相对关系。更准确地说，如果一行中有一个值 X，我想知道同一列（和组）中有多少其他值在 X-30 内。

也就是说，鉴于此：

DT<-data.table(
X = c(1, 2, 2, 1, 1, 2,  1, 2, 2, 1, 1, 1),
Y = c(100, 101, 133, 134, 150, 156,  190, 200, 201, 230, 233, 234),
Z = c(1, 2, 3, 4, 5, 6,  7, 8, 9, 10, 11, 12))

我想获得一个新列，其中包含值：

N <- c(0, 0, 0, 0, 1, 1,  0, 0, 1, 0, 1, 2)

我尝试了以下方法，但没有得到可以使用的结果：

DT[,list(Y,num=cumsum(Y[-.I]>DT[.I,Y]-30),Z),by=.(X)]

任何想法如何做到这一点？

【问题讨论】：

标签： r data.table

【解决方案1】：

这可能可以通过滚动连接 (?) 来实现，但现在有一个 foverlaps 替代方案

DT[, `:=`(indx = .I, Y2 = Y - 30L, N = 0L)] # Add row index and a -30 interval
setkey(DT, X, Y2, Y) # Sort by X and the intervals (for fovelaps)
res <- foverlaps(DT, DT)[Y2 > i.Y2, .N, keyby = indx] # Run foverlaps and check what can we catch
setorder(DT, indx) # go back to the original order
DT[res$indx, N := res$N][, c("indx", "Y2") := NULL] # update results and remove cols
DT
#     X   Y  Z N
#  1: 1 100  1 0
#  2: 2 101  2 0
#  3: 2 133  3 0
#  4: 1 134  4 0
#  5: 1 150  5 1
#  6: 2 156  6 1
#  7: 1 190  7 0
#  8: 2 200  8 0
#  9: 2 201  9 1
# 10: 1 230 10 0
# 11: 1 233 11 1
# 12: 1 234 12 2

或者，使用foverlaps 的which=TRUE 选项使重叠合并更小：

# as above
DT[, `:=`(indx = .I, Y2 = Y - 30L, N = 0L)]
setkey(DT, X, Y2, Y)

# using which=TRUE:
res <- foverlaps(DT, DT, which=TRUE)[xid > yid, .N, by=xid]
DT[res$xid, N := res$N]
setorder(DT, indx)
DT[, c("Y2","indx") := NULL]

【讨论】：

【解决方案2】：

这是另一种方式：

DT[order(Y), N := 0:(.N-1) - findInterval(Y - 30, Y), by = X]

all.equal(DT$N,N) # TRUE

【讨论】：