R:data.table，将组的第一个和最后一个值设置为 NA

Question

我想将组中的第一个和最后一个值设置为 NA。下面是一个例子:

DT <- data.table(v = rnorm(12), class=rep(1:3, each=4))
DT[, v[c(1,.N)] := NA , by=class]

但这是行不通的。我该怎么做？

Answer 1

目前，解决这个问题的方法是首先提取索引，然后通过引用进行一次赋值。

idx = DT[, .(idx = .I[c(1L, .N)]), by=class]$idx
DT[idx, v := NA]

我会尝试将此示例添加到 Reference semantics小插图。