按索引移动数据表行

Question

在尝试回答 this question on Stack Overflow 时，我遇到了自己的单独问题，我自己无法解决。但是，我真的很想为我的问题找到一个有效的解决方案。我最初将其发布为该问题的答案，但由于它引起了混乱，我认为单独发布此问题是一个更好的决定：

假设我有以下数据：

set.seed(1)
dat <- data.table(value = seq(1,7),value2 = seq(1,7))
shift_id = c(1,2,3,NA,4,NA,5)

我想使用 shift_id 将行移到两个新列中，因此所需的输出是：

   value value2 value1.shift value2.shift
1:     1      1            1            1
2:     2      2            2            2
3:     3      3            3            3
4:     4      4            NA            NA
5:     5      5            4            4
6:     6      6            NA            NA
7:     7      7            5            5

请注意，第 7 行包含第 5 行的内容，因为 shift_id[7]=5 我如何使用 DT 包有效地做到这一点？

非常感谢任何帮助！

Answer 1

你可以试试：

dat[, c("value1.shift", "value2.shift") := list(dat$value[shift_id], dat$value2[shift_id])]

 > dat
   value value2 value1.shift value2.shift
1:     1      1            1            1
2:     2      2            2            2
3:     3      3            3            3
4:     4      4           NA           NA
5:     5      5            4            4
6:     6      6           NA           NA
7:     7      7            5            5

另一种使用数据表特定功能的更通用方法（感谢@David Arenburg 的建议）是：

dat[, paste0(names(dat)[grepl("value", names(dat))], ".shift") := lapply(.SD, `[`, shift_id), .SDcols = grep("value", names(dat))]