如何按指定数量截断R数据框中逗号分隔字符串的末尾元素？

现有数据框如下，其中Area_bsl列存储逗号分隔的字符串，diffr列指定需截断该列末尾的元素数量：

df <- data.frame(
  id = 1:3,
  Area_bsl = c("155,199,198,195,100,112,177,199,188,144",
               "100,99,98,95,100,112,111,99",                        
               "131,166,155,111,100,117,166,188,101,101,105,166"),
  diffr = c(3,0,6)
)

需求：

• id == 1时，截断Area_bsl最后3个元素
• id == 2时，不截断（截断0个元素）
• id == 3时，截断Area_bsl最后6个元素

尝试用tidyverse处理，但slice_head步骤报错：

library(tidyverse)
df %>%
  # 将逗号分隔值拆分为行：
  separate_rows(Area_bsl) %>%
  # 按`id`分组：
  group_by(id) %>%
  # 创建行计数器：
  mutate(rowid = row_number()) %>%
  # 创建截断点：
  mutate(cutoff = last(rowid) - diffr) %>%
  # 截取`cutoff`指定数量的行：<--- 执行失败！
  slice_head(n = cutoff[1])

报错信息：

Error in `slice_head()`:
! `n` must be a constant.
Caused by error in `force()`:
! object 'cutoff' not found

期望得到如下结果：

id Area_bsl diffr rowid cutoff
   <int> <chr>    <dbl> <int>  <dbl>
 1     1 155          3     1      7
 2     1 199          3     2      7
 3     1 198          3     3      7
 4     1 195          3     4      7
 5     1 100          3     5      7
 6     1 112          3     6      7
 7     1 177          3     7      7
11     2 100          0     1      8
12     2 99           0     2      8
13     2 98           0     3      8
14     2 95           0     4      8
15     2 100          0     5      8
16     2 112          0     6      8
17     2 111          0     7      8
18     2 99           0     8      8
19     3 131          6     1      6
20     3 166          6     2      6
21     3 155          6     3      6
22     3 111          6     4      6
23     3 100          6     5      6
24     3 117          6     6      6

报错原因 #

slice_head()的n参数要求传入常量值，无法直接引用分组内计算出的变量（比如cutoff）作为动态参数，这是导致报错的核心原因。

解决方案 #

方法1：用filter替代slice_head #

利用已生成的rowid和cutoff列直接筛选行，避开slice_head的常量限制：

library(tidyverse)

df %>%
  separate_rows(Area_bsl) %>%
  group_by(id) %>%
  mutate(rowid = row_number()) %>%
  mutate(cutoff = last(rowid) - diffr) %>%
  filter(rowid <= cutoff) %>%
  ungroup() # 可选，按需取消分组

方法2：用slice动态计算行范围 #

无需单独创建cutoff列，直接在slice中计算需保留的行号范围：

df %>%
  separate_rows(Area_bsl) %>%
  group_by(id) %>%
  mutate(rowid = row_number()) %>%
  slice(1:(n() - first(diffr))) %>%
  mutate(cutoff = n()) %>% # 若需要保留cutoff列则添加此步
  ungroup()

结果说明 #

两种方法均能生成符合需求的结果：

• 方法1通过filter直接匹配行号与截断条件，逻辑直观；
• 方法2利用n()获取分组总行数，结合first(diffr)直接计算保留范围，代码更简洁。

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