1. Învăţa
    2. /
  2. Courses
    3. /
  3. Rcpp로 R 코드 최적화하기
Connected

exercise

Mean carried forward

마지막 관측값 보간(LOCF)의 다른 방법으로, NA를 이전의 결측이 아닌 값들의 평균으로 대체할 수 있어요. 이를 mean carried forward라고 합니다. 여기서도 R에서는 가독성과 속도 사이에서 선택해야 해요. 아래 코드는 가독성을 위해 작성된 버전입니다:

na_meancf1 <- function(x) {
  total_not_na <- 0
  n_not_na <- 0
  res <- x
  for(i in seq_along(x)) {
    if(is.na(x[i])) {
      res[i] <- total_not_na / n_not_na
    } else {
      total_not_na <- total_not_na + x[i]
      n_not_na <- n_not_na + 1
    }
  }
  res
}

반복 계산의 특성 때문에 벡터화가 까다롭습니다. 대신 C++로 변환해 보죠. na_meancf1()을 C++로 옮긴 na_meancf2() 정의를 완성하세요.

Instrucţiuni

100 XP
  • if 조건에서 x의 i번째 요소가 NumericVector의 NA인지 확인하세요.
  • 조건이 참이면, 결과의 i번째 값을 결측이 아닌 값들의 합 total_not_na를 결측이 아닌 값의 개수 n_not_na로 나눈 값으로 설정하세요.
  • 그렇지 않다면 total_not_na에 x의 i번째 요소를 더하고, n_not_na에는 1을 더하세요.