Início / coding / Perguntas / 79048695
Accepted
Ben_89
Asked: 2024-10-03 07:04:10 +0800 CST

Como modificar a largura da janela na função rollapplyr() com base nos nomes das colunas em R

  • 772

Tenho um conjunto de dados que contém várias variáveis ​​de interesse. O que eu gostaria de fazer é calcular uma média móvel com a rollapplyr()função do zoopacote para cada uma delas como colunas adicionais, mas a ressalva é que eu gostaria de usar uma largura de janela diferente ( width =) dependendo da coluna. Também desejo evitar pivotar meus dados para o formato longo.

library(tidyverse)
library(zoo)

rolling_f <- function(x, wdw, fun){ rollapplyr(x, wdw, fun, partial = TRUE) }

set.seed(1)
ex <- data.frame(
  var1 = rnorm(20, 8000, 500),
  var2 = rnorm(20, 8000, 500),
  var3 = rnorm(20, 8000, 500),
  var4 = rnorm(20, 8000, 500),
  var5 = rnorm(20, 8000, 500),
  var6 = rnorm(20, 8000, 500),
  var7 = rnorm(20, 8000, 500),
  var8 = rnorm(20, 8000, 500)
  ) %>%
  mutate(
    across(
      .cols = starts_with("var"),
      .fns = ~rolling_f(.x, 5, mean),
      .names = "{.col}_roll"
      )
    )

Meu código até agora calcula a mesma largura de janela (5) para cada coluna, mas espero que alguém possa me ajudar a definir minha função personalizada melhor para que a janela possa ser 3 para var1:var3, 5 para var4:var5, e 6 para var6:var8, como um exemplo. Presumo que isso simplesmente exigirá algum código adicional na rolling_f()função personalizada que comecei.

2 respostas

  1. Best Answer

    Você pode usar case_match()/ case_when()em relação ao nome da coluna atual em across para determinar o tamanho da janela:

    library(dplyr)
library(readr)
library(zoo)

rolling_f <- function(x, wdw, fun){ rollapplyr(x, wdw, fun, partial = TRUE) }

# case_match()
ex %>%
  mutate(across(
    .cols = starts_with("var"),
    .fns = ~ rolling_f(.x, case_match(
      parse_number(cur_column()), 1:3 ~ 3, 4:5 ~ 5, 6:8 ~ 6
    ), mean),
    .names = "{.col}_roll"
  ))

# case_when()
ex %>%
  mutate(across(
    .cols = starts_with("var"),
    .fns = ~ rolling_f(
      .x,
      case_when(
        cur_column() %in% paste0("var", 1:3) ~ 3,
        cur_column() %in% paste0("var", 4:5) ~ 5,
        cur_column() %in% paste0("var", 6:8) ~ 6
      ),
      mean
    ),
    .names = "{.col}_roll"
  ))
    • 3

  2. Você também pode usar data.table::set(), com um vetor nomeado:

    
library(data.table)
setDT(ex)
m = setNames(c(3,3,3,5,5,6,6,6), paste0("var",1:8))
for(i in names(m)) set(ex,j=paste0(i,"_roll"),value=rolling_f(ex[[i]],m[i],mean))
    • 2

relate perguntas