如何在R语言中按NA分割向量为多个子向量？

针对你的需求，这里提供几种在R中实现按NA分割向量的方法：

方法一：Base R原生实现 #

利用cumsum生成分组标识，结合split完成分割，步骤清晰且无需额外包：

input <- c(1,2,1,NA,3,2,NA,1,5,6,NA,2,2)

# 生成分组ID：每遇到一个NA，分组ID递增
grp <- cumsum(is.na(input))[!is.na(input)] + 1
# 分割非NA元素到对应分组
output <- split(input[!is.na(input)], grp)

# 查看结果
output

运行后会得到预期的列表：

$`1`
[1] 1 2 1

$`2`
[1] 3 2

$`3`
[1] 1 5 6

$`4`
[1] 2 2

方法二：Base R简洁版（含过滤） #

如果想一步完成分割并自动过滤空组，可以结合lapply和Filter：

input <- c(1,2,1,NA,3,2,NA,1,5,6,NA,2,2)

output <- split(input, cumsum(is.na(input))) |>
  lapply(function(x) x[!is.na(x)]) |>
  Filter(f = function(x) length(x) > 0)

方法三：Tidyverse工具链实现 #

如果你习惯使用dplyr和tidyr等tidyverse包，也可以用更直观的管道式操作：

library(tidyverse)

input <- c(1,2,1,NA,3,2,NA,1,5,6,NA,2,2)

output <- tibble(val = input) |>
  # 按NA出现次数生成分组
  mutate(grp = cumsum(is.na(val)) + 1) |>
  # 过滤NA值
  filter(!is.na(val)) |>
  # 按分组拆分列表
  group_split(grp, .keep = FALSE)

原理说明 #

核心逻辑都是通过cumsum(is.na(input))统计每个位置之前出现的NA数量，以此作为分组依据，确保连续的非NA元素被分到同一组，最后提取非NA元素并完成分组拆分。

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