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插图中。

Answer 2

这可能不是单行，但它在代码中确实有“第一个”和“最后一个”：）

> DT <- data.table(v = rnorm(12), class=rep(1:3, each=4))
> setkey(DT, class)
> classes = DT[, .(unique(class))]
> DT[classes, v := NA, mult='first']
> DT[classes, v := NA, mult='last']
> DT
          v class
 1:      NA     1
 2: -1.8191     1
 3: -0.6355     1
 4:      NA     1
 5:      NA     2
 6: -1.1771     2
 7: -0.8125     2
 8:      NA     2
 9:      NA     3
10:  0.2357     3
11:  0.3416     3
12:      NA     3
>

非键列也保留了订单。我认为这是一个记录（承诺）的功能。

Answer 3

使用辅助功能很容易

set.na = function(x,y) {x[y] = NA; x}
DT[, set.na(v,c(1,.N)) , by=class]

Answer 4

修改数据子集的规范方法是使用i来定义子集。您无法将[与:=一起使用。根据@David Arenburg的建议创建临时i，或者您可以使用c(NA, v[-c(1, .N)], NA)构造自己创建结果向量。

DT[, v := c(NA, v[-c(1, .N)], NA)[1:.N], by = class]

但是，您还应注意，例如，您可以更改行顺序。设置新密钥或使用任意数量的功能。所以你应该非常小心这个操作。

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

4 个答案: