在数据框的每一行上选择不同数量的列

Question

这个问题是关于在数据框的每一行上选择不同数量的列。我有一个数据框：

df = data.frame(
    START=sample(1:2, 10, repace=T), END=sample(2:4, 10, replace=T),
    X1=rnorm(10), X2=rnorm(10), X3=rnorm(10), X4=rnorm(10)
)

我希望有一种不使用循环的方法来为我的数据框的所有行选择第 i 行上的列 (START[i]:END[i])+2。

Answer 1

基础 R 解决方案

lapply(split(df,1:nrow(df)),function(row) row[(row$START+2):(row$END+2)])

或者类似于上面评论中给出的东西（我会将输出存储在一个列表中）

library(plyr)
alply(df,1,function(row) row[(row$START+2):(row$END+2)])

---

根据 OP 的请求进行编辑：

要获得 TRUE/FALSE 索引矩阵，请使用以下 R 基础解决方案

idx_matrix=col(df)>=df$START+2&col(df)<=df$END+2
df[idx_matrix]

但是请注意，您会在此处丢失一些信息（与基于列表的解决方案相比）。