data.frame (또는 data.table)에서 NA에 가장 가까운 이전의 비 NA 값으로 "채우기"하고 싶습니다. 간단한 대신에 벡터를 사용하는 예 data.frame
는 다음과 같습니다.
> y <- c(NA, 2, 2, NA, NA, 3, NA, 4, NA, NA)
나는 다음과 같이 fill.NAs()
구성 할 수 있는 기능 을 원합니다 yy
.
> yy
[1] NA NA NA 2 2 2 2 3 3 3 4 4
data.frame
행이 NA 인 모든 항목이 많은 (총 ~ 1 Tb) 작은 크기 (~ 30-50 Mb)에 대해이 작업을 반복해야합니다 . 문제에 접근하는 좋은 방법은 무엇입니까?
내가 요리 한 못생긴 해결책은이 기능을 사용합니다.
last <- function (x){
x[length(x)]
}
fill.NAs <- function(isNA){
if (isNA[1] == 1) {
isNA[1:max({which(isNA==0)[1]-1},1)] <- 0 # first is NAs
# can't be forward filled
}
isNA.neg <- isNA.pos <- isNA.diff <- diff(isNA)
isNA.pos[isNA.diff < 0] <- 0
isNA.neg[isNA.diff > 0] <- 0
which.isNA.neg <- which(as.logical(isNA.neg))
if (length(which.isNA.neg)==0) return(NULL) # generates warnings later, but works
which.isNA.pos <- which(as.logical(isNA.pos))
which.isNA <- which(as.logical(isNA))
if (length(which.isNA.neg)==length(which.isNA.pos)){
replacement <- rep(which.isNA.pos[2:length(which.isNA.neg)],
which.isNA.neg[2:max(length(which.isNA.neg)-1,2)] -
which.isNA.pos[1:max(length(which.isNA.neg)-1,1)])
replacement <- c(replacement, rep(last(which.isNA.pos), last(which.isNA) - last(which.isNA.pos)))
} else {
replacement <- rep(which.isNA.pos[1:length(which.isNA.neg)], which.isNA.neg - which.isNA.pos[1:length(which.isNA.neg)])
replacement <- c(replacement, rep(last(which.isNA.pos), last(which.isNA) - last(which.isNA.pos)))
}
replacement
}
이 기능 fill.NAs
은 다음과 같이 사용됩니다.
y <- c(NA, 2, 2, NA, NA, 3, NA, 4, NA, NA)
isNA <- as.numeric(is.na(y))
replacement <- fill.NAs(isNA)
if (length(replacement)){
which.isNA <- which(as.logical(isNA))
to.replace <- which.isNA[which(isNA==0)[1]:length(which.isNA)]
y[to.replace] <- y[replacement]
}
산출
> y
[1] NA 2 2 2 2 3 3 3 4 4 4
... 작동하는 것 같습니다. 그러나, 사람, 그것은 추악하다! 어떤 제안?
fill
에서R
tidyr::fill()
.
roll=TRUE
에data.table
.