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语义插图中.