有什么功能可以从R中的数据帧中提取几行(不连续)?

时间:2019-10-02 11:50:37

标签: r dataframe

我正在尝试从数据帧中提取几行(不是一个接一个)。应该删除的行范围在另一个数据框中。

我尝试使用for循环提取行,但不幸的是,仅删除了最后一个范围。

这是我使用的代码行(在“ for”循环中,而“ i”是变量):

new_df <- main_df[-(erase_df$starts[i]:erase_df$stops[i]),]

例如: 这是我要更改的数据框(main_df)

> main_df
   v1  v2     v3
1   1 bla blabla
2   2 bla blabla
3   3 bla blabla
4   4 bla blabla
5   5 bla blabla
6   6 bla blabla
7   7 bla blabla
8   8 bla blabla
9   9 bla blabla
10 10 bla blabla
11 11 bla blabla
12 12 bla blabla
13 13 bla blabla
14 14 bla blabla
15 15 bla blabla

这是包含我要删除的行范围的数据帧(erase_df)(“开始”矢量表示应删除第一个行范围,“停止”矢量表示应删除最后一行)范围)

> erase_df
  starts stops
1      3     5
2      9    10
3     12    14

因此新数据框应如下所示:

> new_df
   v1  v2     v3
1   1 bla blabla
2   2 bla blabla
6   6 bla blabla
7   7 bla blabla
8   8 bla blabla
11 11 bla blabla
15 15 bla blabla

我希望输出看起来像我上面提到的(new_df),但取而代之的是,仅删除了delete_df中的最后一个范围(开始= 12,停止= 14)

1 个答案:

答案 0 :(得分:0)

如果您在Mapseq函数erase_df中创建一行要删除的行序列,然后unlist将它们全部合并为一个向量,则可以具有该向量的负数的main_df子集,以删除给定范围内的行。

remove <- unlist(Map(seq, erase_df[[1]], erase_df[[2]]))

main_df[-remove,]
#    v1  v2     v3
# 1:  1 bla blabla
# 2:  2 bla blabla
# 3:  6 bla blabla
# 4:  7 bla blabla
# 5:  8 bla blabla
# 6: 11 bla blabla
# 7: 15 bla blabla

或者,对于一个更复杂的选项,该选项对于更大的数据可能更有效(未经测试,只是猜测)

library(data.table)
setDT(main_df)
setDT(erase_df)

setkey(erase_df, starts, stops)
main_df[, v0 := v1]
for_anti <- 
  foverlaps(main_df, erase_df, by.x = c('v0', 'v1'), type = 'within',
            nomatch = NULL)

main_df[!for_anti, on = .(v1)]
#    v1  v2     v3 v0
# 1:  1 bla blabla  1
# 2:  2 bla blabla  2
# 3:  6 bla blabla  6
# 4:  7 bla blabla  7
# 5:  8 bla blabla  8
# 6: 11 bla blabla 11
# 7: 15 bla blabla 15