学习 data.table - 如何按行号和列名更新值

Question

我最近的几篇文章写得不好，所以这次我会尝试做得更好，更干净。

我正在学习如何使用数据表对象，而我正在努力完成的一项任务是同时按行号和列名更新数据表中的值。使用 data.frames 这会容易得多，我只需执行以下操作：

my_df = as.data.frame(matrix(ncol = 10, nrow = (100)))
names(my_df) = c("P1", "P2", "P3", "P4", "P5", "Q1", "Q2", "Q3", "Q4", "Q5")
head(my_df)

  P1 P2 P3 P4 P5 Q1 Q2 Q3 Q4 Q5
1 NA NA NA NA NA NA NA NA NA NA
2 NA NA NA NA NA NA NA NA NA NA
3 NA NA NA NA NA NA NA NA NA NA
4 NA NA NA NA NA NA NA NA NA NA
5 NA NA NA NA NA NA NA NA NA NA
6 NA NA NA NA NA NA NA NA NA NA

replacement = c(1, 2, 3, 4, 5)
my_df[2, names(my_df)[1:5]] = replacement
head(my_df)

  P1 P2 P3 P4 P5 Q1 Q2 Q3 Q4 Q5
1 NA NA NA NA NA NA NA NA NA NA
2  1  2  3  4  5 NA NA NA NA NA
3 NA NA NA NA NA NA NA NA NA NA
4 NA NA NA NA NA NA NA NA NA NA
5 NA NA NA NA NA NA NA NA NA NA
6 NA NA NA NA NA NA NA NA NA NA

所以，使用数据框相当容易。但是，我正在努力使用数据表完成同样的任务。对示例数据表使用与上面的数据框相同的结构，我尝试了以下方法：

my_dt = data.table(matrix(ncol = 10, nrow = (100)))
names(my_dt) = c("P1", "P2", "P3", "P4", "P5", "Q1", "Q2", "Q3", "Q4", "Q5")
head(my_dt)

   P1 P2 P3 P4 P5 Q1 Q2 Q3 Q4 Q5
1: NA NA NA NA NA NA NA NA NA NA
2: NA NA NA NA NA NA NA NA NA NA
3: NA NA NA NA NA NA NA NA NA NA
4: NA NA NA NA NA NA NA NA NA NA
5: NA NA NA NA NA NA NA NA NA NA
6: NA NA NA NA NA NA NA NA NA NA

replacement = c(1, 2, 3, 4, 5)
# my_dt[i == 2, names(my_dt)[1:5]] = replacement
# my_dt[i == 2, names(my_dt)[1:5] := replacement]  
# my_dt[2, names(my_dt)[1:5]] = replacement
# my_dt[2, names(my_dt)[1:5] := replacement]  

然而，四个注释行都没有正确替换。感谢您的帮助！

谢谢， 卡诺维斯

Answer 1

或者你可以这样做：

x <- names(my_dt)[1:5]

my_dt[, (x) := lapply(.SD, as.numeric), .SDcols = x]

my_dt[2,  (x):= as.list(replacement)]

首先我们将my_dt 中的目标列转换为numeric。 .SDcols 表示.SD 中我们感兴趣的列的子集。.SD 包含data.table 中的所有列（by 中使用的列除外）。

将目标列转换为数字后，我们会通过引用更新值。

注意：不必事先定义 x，一切都可以即时完成。但是，如果您定义x，则需要将其包装在() 中以确保data.table 不会查找x 列

Answer 2

对您的示例进行了一些小的更改，但这有效：

#Filled data.table with integers instead of NAs to avoid converting 
#from logical later
#Left out names as it wasn't relevant to the example
my_dt = as.data.table(matrix(ncol = 10, nrow = (100), 1L))
head(my_dt)

replacement <- 1:5
#Loop through columns and use set to replace values without making a copy
for(k in 1:5) set(my_dt, i = 2L, j = k , value = replacement[k])
head(my_dt)