具有多列的数据表中的子集行

Question

我正在尽力不使用 data.table 计算行，因为这是不可取的。我想对具有许多列的大型 data.table 的值大于或等于 2 的所有行进行子集化，然后可能还会获取这些列。

可重现的例子：

 DT <- as.data.table( rbind( diag(5)+1, diag(5), rep(0,5)) )
 DT[1,1] = 1

我的代码：

DT[,lapply(.SD, function(x) x>=2 )]

我得到一个 data.table 的布尔值，现在很困惑我应该如何子集化它。

@IRTFM 提到的一种可行的方法是：

DT[DT[,apply(.SD, 1, function(x) any(x>=2))]]

    V1 V2 V3 V4 V5
1:  1  2  1  1  1
2:  1  1  2  1  1
3:  1  1  1  2  1
4:  1  1  1  1  2

现在我应该如何对具有2 值的列进行子集化？？

理想的结果是：

      V2 V3 V4 V5
1:    2  1  1  1
2:    1  2  1  1
3:    1  1  2  1
4:    1  1  1  2

Answer 1

我无法让您的示例加载到我的控制台会话中，但这是一个更“最小值”；演示方法的示例。不确定它是否具有通常的 data.table 效率：

DT <- setDT( data.frame(x=1:2, y=0,z=0))
DT[, apply(.SD, 1, function(x){any(x>=2)}) ] gets you a logical vector for each row
# [1] FALSE  TRUE
DT[ DT[, apply(.SD, 1, function(x){any(x>=2)}) ]] # uses that vector to select rows
   x y z
1: 2 0 0

这也应该成功：

DT[ as.logical(rowSums(DT >= 2))]
   x y z
1: 2 0 0

对于第二部分，请考虑：

cols <- sapply(DT, function(x){ any(x>0)})
DT2[ ,.SD, .SDcols=names(cols[cols])]