如何使用(row,col)索引序列对data.table进行切片取值？

嗨，针对你想用原生data.table语法高效提取对应位置元素的需求，这里有几个不错的解决方案，完全不用转成data.frame或者写显式循环：

方法1：转成矩阵后用矩阵索引（高效简洁） #

这是最直接且性能优异的方式，利用R原生的矩阵索引特性，data.table转矩阵的开销非常小，尤其是处理大数据集时表现出色：

# 定义提取函数
get_elements <- function(ri, ci, tab) {
  as.matrix(tab)[cbind(ri, ci)]
}

# 测试你的对角线例子
ri <- 1:3
ci <- 1:3
get_elements(ri, ci, tab)
# 输出：NA 42190 NA

方法2：纯data.table原生语法（无需转结构） #

如果你想完全依赖data.table的操作逻辑，可以通过行索引+列索引筛选后取对角线元素的方式实现——我们筛选后的表中第i行第i列，正好对应tab[ri[i], ci[i]]的目标元素：

# 定义提取函数
get_elements <- function(ri, ci, tab) {
  diag(tab[ri, ci, with = FALSE])
}

# 测试自定义索引的情况
ri <- c(1, 2, 3)
ci <- c(3, 1, 2)
get_elements(ri, ci, tab)
# 输出：40570 42190 NA

为什么这两种方法更好？ #

• 避免了as.data.frame(tab)[cbind(ri, ci)]这种依赖data.frame的转换操作，更贴合data.table的使用场景
• 完全没有显式for循环，性能远高于循环实现，尤其是数据量较大时
• 代码简洁易读，维护成本低

内容的提问来源于stack exchange，提问作者Joshua Daly