R 将函数应用于 data.table 中的每个指定列并通过引用更新

发布于05月31日

我有数据.我希望对某些列执行相同操作的表.这些列的名称在字符向量中给出.在这个特定的例子中，我想将所有这些列乘以-1.

一些玩具数据和指定相关列的向量:

library(data.table)
dt <- data.table(a = 1:3, b = 1:3, d = 1:3)
cols <- c("a", "b")

现在我这样做，在角色向量上循环:

for (col in 1:length(cols)) {
   dt[ , eval(parse(text = paste0(cols[col], ":=-1*", cols[col])))]
}

有没有一种方法可以不使用for循环直接实现这一点？

推荐答案

这似乎有效:

dt[ , (cols) := lapply(.SD, "*", -1), .SDcols = cols]

结果是

    a  b d
1: -1 -1 1
2: -2 -2 2
3: -3 -3 3

这里有一些技巧:

因为(cols) :=中有括号，所以结果被分配给cols中指定的列，而不是一些名为"cols"的新变量.
.SDcols告诉调用我们只查看这些列，并允许我们使用.SD，即与这些列关联的Data的S子集.
lapply(.SD, ...)在.SD上运行，.SD是一个列列表(如所有data.frames和data.tables).lapply返回一个列表，所以最后j看起来像cols := list(...).

EDIT:这里有另一种可能更快的方法，正如@Arun提到的:

for (j in cols) set(dt, j = j, value = -dt[[j]])