В векторе, содержащем блоки чисел и блоки NA, например:
score <- c(0,1,2,3,4,NA,NA,0,-1,0,1,2,NA,NA,NA)
есть ли способ имитировать пропущенные значения путем подсчета вверх по шагам от последнего значения до блока NA?
Таким образом, это будет:
score.correct <- c(0,1,2,3,4,5,6,0,-1,0,1,2,3,4,5)
Спасибо за любую помощь.
Q+D, имеет цикл, делает некоторые ненужные добавления, но выполняет работу:
incna <- function(s){
while(any(is.na(s))){
ina = which(is.na(s))
s[ina]=s[ina-1]+1
}
s
}
> score
[1] 0 1 2 3 4 NA NA 0 -1 0 1 2 NA NA NA
> incna(score)
[1] 0 1 2 3 4 5 6 0 -1 0 1 2 3 4 5
Сбой с только предупреждением, если первым элементом является NA:
> score
[1] NA 1 2 3 4 NA NA 0 -1 0 1 2 NA NA NA
> incna(score)
[1] 5 1 2 3 4 5 3 0 -1 0 1 2 3 4 5
Warning message:
In s[ina] = s[ina - 1] + 1 :
number of items to replace is not a multiple of replacement length
Взято из Кристоса Хатзиса на r-help:
rna <- function(z) {
y <- c(NA, head(z, -1))
z <- ifelse(is.na(z), y+1, z)
if (any(is.na(z))) Recall(z) else z }
rna(score)
#[1] 0 1 2 3 4 5 6 0 -1 0 1 2 3 4 5
Драконы:
rna(c(NA,score))
Error: evaluation nested too deeply: infinite recursion / options(expressions=)?
rna(c(1,rep(NA,1e4)))
Error: evaluation nested too deeply: infinite recursion / options(expressions=)?
Ориентир:
score2 <- 1:1e5
set.seed(42)
score2[sample(score2,10000)] <- NA
library(microbenchmark)
microbenchmark(rna(score2),incna(score2))
Unit: milliseconds
expr min lq median uq max
1 incna(score2) 2.93309 2.973896 2.990988 3.134501 5.360186
2 rna(score2) 50.42240 50.848931 51.228040 52.778043 56.856773
NAs. Попробуйте score <- c(1, rep(NA, 1000))... 01.02.2013 Вот еще один подход:
library(zoo)
ifelse(is.na(score), na.locf(score) + sequence(rle(is.na(score))$l), score)
# [1] 0 1 2 3 4 5 6 0 -1 0 1 2 3 4 5
Показаны промежуточные результаты, где [] указывает на NA слотов:
na.locf(score)
# [1] 0 1 2 3 4 [4] [4] 0 -1 0 1 2 [2] [2] [2]
sequence(rle(is.na(score))$l)
# [1] 1 2 3 4 5 [1] [2] 1 2 3 4 5 [1] [2] [3]
na.locf(score) + sequence(rle(is.na(score))$l)
# [1] 1 3 5 7 9 [5] [6] 1 1 3 5 7 [3] [4] [5]
ifelse(is.na(score), na.locf(score) + sequence(rle(is.na(score))$l), score)
# [1] 0 1 2 3 4 [5] [6] 0 -1 0 1 2 [3] [4] [5]
