R data.table - 根据值添加行

Question

下面有data.table x

x <- data.table(id=c('A1', 'B1'), start=c(1,1), stop=c(4,5))

id	start	stop
A1	1	4
B1	1	5

我想扩展行。是否可以使用 rbindlist 使用 Map 生成 data.table 如下？

id	start	stop
A1	1	2
A1	2	3
A1	3	4
B1	1	2
B1	2	3
B1	3	4
B1	4	5

Answer 1

您可以为每个id 创建一个从start 到stop 的序列。使用shift 获取下一个值并删除NA 行。

library(data.table)
x <- x[, .(start = seq(start, stop)), id]
x[, stop := shift(start, type = 'lead'), id]
x[!is.na(stop)]

#   id start stop
#1: A1     1    2
#2: A1     2    3
#3: A1     3    4
#4: B1     1    2
#5: B1     2    3
#6: B1     3    4
#7: B1     4    5

这是一个等效的 tidyverse 方式 -

library(tidyverse)

x %>%
  mutate(start = map2(start, stop, seq)) %>%
  unnest(start) %>%
  group_by(id) %>%
  mutate(stop = lead(start)) %>%
  ungroup %>%
  filter(!is.na(stop))