Расчет скользящей средней

r moving-average r-faq

194250 просмотра

9 ответа

28564 Репутация автора

Я пытаюсь использовать R для расчета скользящего среднего по ряду значений в матрице. Обычный поиск по списку рассылки R не очень помог. Кажется, в R нет встроенной функции, которая позволяла бы мне вычислять скользящие средние. Какие-нибудь пакеты предоставляют? Или мне нужно написать свое?

Автор: Jared Источник Размещён: 13.04.2009 12:52

Ответы (9)


120 плюса

23926 Репутация автора

Решение Автор: f3lix Размещён: 13.04.2009 01:06

186 плюса

7059 Репутация автора

Или вы можете просто рассчитать его с помощью фильтра, вот функция, которую я использую:

ma <- function(x,n=5){filter(x,rep(1/n,n), sides=2)}

Автор: Matti Pastell Размещён: 01.02.2011 12:06

6 плюса

42456 Репутация автора

caToolsПакет очень быстро прокатки среднего / мин / макс / SD и несколько других функций. Я только работал с, runmeanи runsdони являются самыми быстрыми из всех других пакетов, упомянутых на сегодняшний день.

Автор: eddi Размещён: 21.08.2013 05:11

8 плюса

9826 Репутация автора

Вы можете использовать RcppRollдля очень быстрых скользящих средних, написанных на C ++. Просто вызовите roll_meanфункцию. Документы можно найти здесь .

В противном случае этот (более медленный) цикл for должен сработать:

ma <- function(arr, n=15){
  res = arr
  for(i in n:length(arr)){
    res[i] = mean(arr[(i-n):i])
  }
  res
}
Автор: cantdutchthis Размещён: 16.01.2015 02:49

24 плюса

361 Репутация автора

Использование cumsumдолжно быть достаточным и эффективным. Предполагая, что у вас есть вектор x, и вы хотите получить текущую сумму из n чисел

cx <- c(0,cumsum(x))
rsum <- (cx[(n+1):length(cx)] - cx[1:(length(cx) - n)]) / n

Как отмечено в комментариях @mzuther, это предполагает, что в данных нет NA. чтобы справиться с этим, потребуется разделить каждое окно на количество значений, отличных от NA. Вот один из способов сделать это, включив комментарий @Ricardo Cruz:

cx <- c(0, cumsum(ifelse(is.na(x), 0, x)))
cn <- c(0, cumsum(ifelse(is.na(x), 0, 1)))
rx <- cx[(n+1):length(cx)] - cx[1:(length(cx) - n)]
rn <- cn[(n+1):length(cx)] - cn[1:(length(cx) - n)]
rsum <- rx / rn

Это все еще имеет проблему, что, если все значения в окне являются NA, тогда будет ошибка деления на ноль.

Автор: pipefish Размещён: 12.08.2015 08:26

7 плюса

430 Репутация автора

На самом деле RcppRollэто очень хорошо.

Код, размещенный cantdutch, должен быть исправлен в четвертой строке, чтобы окно было исправлено:

ma <- function(arr, n=15){
  res = arr
  for(i in n:length(arr)){
    res[i] = mean(arr[(i-n+1):i])
  }
  res
}

Другой способ, который обрабатывает пропуски, дан здесь .

Третий способ - улучшить этот код для вычисления частичных средних значений или нет - следующим образом:

  ma <- function(x, n=2,parcial=TRUE){
  res = x #set the first values

  if (parcial==TRUE){
    for(i in 1:length(x)){
      t<-max(i-n+1,1)
      res[i] = mean(x[t:i])
    }
    res

  }else{
    for(i in 1:length(x)){
      t<-max(i-n+1,1)
      res[i] = mean(x[t:i])
    }
    res[-c(seq(1,n-1,1))] #remove the n-1 first,i.e., res[c(-3,-4,...)]
  }
}
Автор: Rodrigo Remedio Размещён: 13.03.2016 02:02

4 плюса

527 Репутация автора

Для того, чтобы дополнить ответ cantutchthis и Родриго Ремедио ;

moving_fun <- function(x, w, FUN, ...) {
  # x: a double vector
  # w: the length of the window, i.e., the section of the vector selected to apply FUN
  # FUN: a function that takes a vector and return a summarize value, e.g., mean, sum, etc.
  # Given a double type vector apply a FUN over a moving window from left to the right, 
  #    when a window boundary is not a legal section, i.e. lower_bound and i (upper bound) 
  #    are not contained in the length of the vector, return a NA_real_
  if (w < 1) {
    stop("The length of the window 'w' must be greater than 0")
  }
  output <- x
  for (i in 1:length(x)) {
     # plus 1 because the index is inclusive with the upper_bound 'i'
    lower_bound <- i - w + 1
    if (lower_bound < 1) {
      output[i] <- NA_real_
    } else {
      output[i] <- FUN(x[lower_bound:i, ...])
    }
  }
  output
}

# example
v <- seq(1:10)

# compute a MA(2)
moving_fun(v, 2, mean)

# compute moving sum of two periods
moving_fun(v, 2, sum)
Автор: Cristóbal Alcázar Размещён: 29.06.2017 02:22

0 плюса

1 Репутация автора

Хотя и немного медленно, но вы также можете использовать zoo :: rollapply для выполнения вычислений на матрицах.

reqd_ma <- rollapply(x, FUN = mean, width = n)

где x - набор данных, FUN = mean - функция; Вы также можете изменить его на min, max, sd и т. д., а width - это скользящее окно.

Автор: Garima gulati Размещён: 11.09.2018 04:34

5 плюса

7835 Репутация автора

В data.table 1.12.0 новая frollmeanфункция была добавлена для вычисления быстрого и точного прокатки означают тщательно обработки NA, NaNи +Inf, -Infзначения.

Так как в этом вопросе нет воспроизводимого примера, здесь не так много вопросов.

Вы можете найти более подробную информацию о ?frollmeanв руководстве, также доступны в Интернете по адресу ?frollmean.

Примеры из руководства ниже:

library(data.table)
d = as.data.table(list(1:6/2, 3:8/4))

# rollmean of single vector and single window
frollmean(d[, V1], 3)

# multiple columns at once
frollmean(d, 3)

# multiple windows at once
frollmean(d[, .(V1)], c(3, 4))

# multiple columns and multiple windows at once
frollmean(d, c(3, 4))

## three above are embarrassingly parallel using openmp
Автор: jangorecki Размещён: 22.12.2018 03:00
Вопросы из категории :
32x32