r 使用带有 NA 的行来 reshape 数据以识别新列

Question

我在 R 中有一个数据集，如下所示:

DF <- data.frame(name=c("A","b","c","d","B","e","f"),
                 x=c(NA,1,2,3,NA,4,5))

我想把它改造成:

rDF <- data.frame(name=c("b","c","d","e","f"),
                  x=c(1,2,3,4,5),
                  head=c("A","A","A","B","B"))

其中第一行带有 NA标识一个新列，并使用该“行值”直到下一行带有 NA ，然后更改“行值”。

我都试过了 spread和 melt ，但它没有给我我想要的。

library(tidyr)
DF %>% spread(name,x)
library(reshape2)
melt(DF, id=c('name'))

有什么建议？

Answer 1

这是一个可能的 data.table/zoo套餐组合解决方案

library(data.table) ; library(zoo)

setDT(DF)[is.na(x), head := name]
na.omit(DF[, head := na.locf(head)], "x")

#    name x head
# 1:    b 1    A
# 2:    c 2    A
# 3:    d 3    A
# 4:    e 4    B
# 5:    f 5    B

或者按照@Arun 的建议，只需使用 data.table

na.omit(setDT(DF)[, head := name[is.na(x)], by=cumsum(is.na(x))])