R语言split函数能否直接提取拆分后数据框的前N个分组

基础R中的split()函数没有内置对应参数，无法通过传参直接实现“拆分时仅保留前N个/后N个分组”的效果。

split()的核心设计逻辑是根据传入分组变量的所有出现水平，将输入数据完整拆分为一一对应的列表结果，它的官方内置参数仅包含：待拆分对象x、分组规则f、是否丢弃空水平的drop、控制分组名拼接的sep/lex等，没有任何参数支持拆分过程中自动筛选保留指定数量的分组，你设想的传参调用方式无法原生支持。

你可以用两种非常简单的写法实现仅保留0、1两个分组的需求，代码改动量极小：

方法1：拆分后按索引取子集（改动最小，适配你的现有代码） #

findInterval()返回的分组值是从0开始递增的连续整数，split()返回的列表默认会按分组值从小到大排序，因此前2个元素固定对应你需要的0、1分组，直接取位置子集即可：

lapply(listdf, function(x) {
  res <- split(x, findInterval(x$Date, dateIntervals))
  res[1:2] # 要取后N个分组替换为tail(res, N)即可
})

注意：如果你的分组值不是有序连续值，不要用位置索引，改用分组名筛选更稳妥，比如写res[c("0","1")]，避免分组顺序变化导致取错数据。

方法2：拆分前先过滤无关行（大数据量下性能更优） #

如果你完全不需要第2组的数据，可以在拆分前就筛掉对应行，避免多余计算：

lapply(listdf, function(x) {
  grp <- findInterval(x$Date, dateIntervals)
  split(x[grp < 2, ], grp[grp < 2])
})

内容的提问来源于stack exchange，提问作者JooHee Lee