R如何仅在所有元素都为零时替换组中的元素

Question

只有当组中的所有元素都为零时，我才想向data.tabledt添加一个新列new_col。

library(data.table)
a <- c(1, 2, 3, 0, 0, 0, 0, 1)
group <- c("a", "a", "a", "b", "b", "c", "c", "c")
dt = data.table(a, group)
dt[]
a group
1: 1     a
2: 2     a
3: 3     a
4: 0     b
5: 0     b
6: 0     c
7: 0     c
8: 1     c

以下是我绑定的代码。

dt[, new_col := ifelse(all(a == 0), NA, a), by = group]
dt[]

我得到的输出是

a group new_col
1: 1     a       1
2: 2     a       1
3: 3     a       1
4: 0     b      NA
5: 0     b      NA
6: 0     c       0
7: 0     c       0
8: 1     c       0

预期输出为

a group new_col
1: 1     a       1
2: 2     a       2
3: 3     a       3
4: 0     b      NA
5: 0     b      NA
6: 0     c       0
7: 0     c       0
8: 1     c       1

有人能指出我做错了什么吗？

Answer 1

这里，我们可以使用if/else，因为ifelse要求所有参数的长度都相同all(a == 0)的长度为1，以及'yes'但'no'的长度不为1，这导致回收

dt[, new_col := if(all(a == 0)) NA else a, by = group]

-输出

dt
a group new_col
1: 1     a       1
2: 2     a       2
3: 3     a       3
4: 0     b      NA
5: 0     b      NA
6: 0     c       0
7: 0     c       0
8: 1     c       1

---

Or也可以使用.I来获得否定表达式的行索引(或使用any(，提取($V1(并将这些元素从"a"分配给new_col。默认情况下，其他元素将是NA

dt[dt[, .I[!all(a == 0)], group]$V1, new_col := a]