return Значение NA в неожиданном поведении NumericVector Rcpp

Question

Я пишу функцию cpp для замены любых значений NA следующим не-na значением. Код работает правильно относительно замены, однако я хочу вернуть значения NA для тех, у которых нет более позднего значения, отличного от NA.

Например:

fill_backward(c(1, NA, 2)) -> 1, 2, 2

fill_backward(c(1, NA, 2, NA)) -> 1, 2, 2, Н. А.

#include <Rcpp.h> 
using namespace Rcpp; 
//' given NA values fill them with the next non-na value 
//' @param x A numeric vector of values 
//' @details 
//' Works very well in context of dplyr to carry out last-observation-carried-foward 
//' for different individuals. It will NOT replace leading NA's 
//' @examples /dontrun { 
//' fill_forward(c(1.0, NA, 2)) 
//' fill_forward(c(NA, 1, NA, 2)) 
//' library(dplyr) 
//' df <- data_frame(id = c(1, 1, 2, 2), obs = c(1.2, 4.8, 2.5, NA)) 
//' df %>% group_by(id) %>% mutate(obs_locf = fill_forward(obs)) 
//' } 
//' @export 
// [[Rcpp::export]] 
NumericVector fill_backward(NumericVector x) { 
    int n = x.size(); 
    NumericVector out = no_init(n); 
    for (int i = 0; i < n; ++i) { 
    if (R_IsNA(x[i])) { 
     for (int j = i+1; j < n; ++j) { 
     if(R_IsNA(x[j])) { 
     continue; 
     } else { 
     out[i] = x[j]; 
     break; 
     } 
     //if never gets to another actual value 
     out[i] = NumericVector::get_na(); 
     } 
    } else { //not NA 
     out[i] = x[i]; 
    } 
    } 
    return out; 
} 

В настоящее время fill_backward(c(NA, 1.0, NA, 2, NA, NA)) возвращается:

[1] 1.000000e+00 1.000000e+00 2.000000e+00 [4] 2.000000e+00 2.156480e-314 -1.060998e-314

вместо 1 1 2 2 NA NA

Для возвращения значения NA назад это out[i] = NumericVector::get_na();

Я также попытался out[i] = REAL_NA и из [я] = х [я] `и ничего не похоже на работу.

Наконец, я использовал реализацию такого же типа для реализации fill_forward, которая может быть видна here, где ведущий NA должен возвращаться как NA - и он правильно возвращает значения NA, поэтому я полностью потеряю.

EDIT: Фиксированный благодаря предложениям @Roland «s

Answer 1

Вы можете инициализировать out с NA значениями:

#include <Rcpp.h> 
using namespace Rcpp; 

// [[Rcpp::export]] 
NumericVector fill_backward(NumericVector x) { 
    int n = x.size(); 
    NumericVector out = NumericVector(n, NumericVector::get_na()); 
    for (int i = 0; i < n; ++i) { 
    if (R_IsNA(x[i])) { 
     for (int j = i+1; j < n; ++j) { 
     if(R_IsNA(x[j])) { 
     continue; 
     } else { 
     out[i] = x[j]; 
     break; 
     } 
      } 
    } else { //not NA 
     out[i] = x[i]; 
    } 
    } 
    return out; 
} 

тестирования:

fill_backward(c(NA, 1.0, NA, 2, NA, NA)) 
[1] 1 1 2 2 NA NA 

И я, вероятно, следует отметить, что ваша линия out[i] = NumericVector::get_na(); никогда не была достигнута из-за использования вами continue.

Answer 2

Пакет zoo уже это делает, и делает это хорошо и быстро:

R> suppressMessages(library(zoo)) 
R> zoo::na.locf(vec, fromLast=TRUE, na.rm=FALSE) 
[1] 1 1 2 2 NA NA 
R> 

и RcppXts пакет позволяет получить доступ зоопарк и XTS код в C++ код, если вы так желаете, включая na.locf функциональность via this access point