r - 使用 := using variable name 更新列

Question

我正在尝试更新 data.table 中的列，但是在使用变量引用列名称时遇到问题:

dt = data.table(a = 1:10, b = 1:10)
nm = 'a'
dt[ , get(nm) := get(nm) + 1]

Error in get(nm) : object 'a' not found

如果我尝试 dt[ , nm := get(nm) + 1] 我会得到一个新列 nm。

我想要的是

     a  b
 1:  2  1
 2:  3  2
 3:  4  3
 4:  5  4
 5:  6  5
 6:  7  6
 7:  8  7
 8:  9  8
 9: 10  9
10: 11 10

Answer 1

这里get返回值。在:=的lhs中，我们只需要提取列名字符串即可。为此，只需用 ()

包裹它

dt[ , (nm) := get(nm) + 1]

从技术上讲，.SDcols 将是指定 data.table 中感兴趣的列的首选方式

dt[, (nm) := .SD[[1]] + 1, .SDcols = nm]

这里，[[用于提取向量。

如果还有更多列，

dt[, (nm) := lapply(.SD, `+`, 1), .SDcols = nm]