通过regex添加data.table列

Question

我想添加data.table的特定列，并希望通过正则表达式选择列。以下面的例子为例。

library(data.table)
mydt <- data.table(
    id = c(1, 2, 3),
    x_0 = c(1, NA, 0), x_1 = c(2, 4, NA), x_2 = c(3, 5, NA), x_3 = c(NA, 6, NA), 
    y_0 = c(6, NA, 0), y_1 = c(5, 3, NA), y_2 = c(4, 2, NA), y_3 = c(NA, 1, NA)
)

mydt
   id x_0 x_1 x_2 x_3 y_0 y_1 y_2 y_3
1:  1   1   2   3  NA   6   5   4  NA
2:  2  NA   4   5   6  NA   3   2   1
3:  3   0  NA  NA  NA   0  NA  NA  NA

我想添加标记为0的列。因此，我希望有一个新列等于x_0，而不是列y_0和x_0 + y_0像这样：

my_preferred_dt
   id x_1 x_2 x_3 y_1 y_2 y_3 new_col
1:  1   2   3  NA   5   4  NA       7
2:  2   4   5   6   3   2   1      NA
3:  3  NA  NA  NA  NA  NA  NA       0

由于我有很多这样的专栏，我想根据正则表达式（"_0$"）选择它们。

Answer 1

您可以将Reduce与SD / SDcols结合使用，只对所需的列进行汇总（<{1}}）：

regex

如果你需要删除＆＃34; ..._ 0＆＃34;变量：

mydt[, new_col := Reduce(`+`, .SD), .SDcols=grep("_0$", names(mydt), value=T)]