Início / coding / Perguntas / 79428166
Accepted
M.O
Asked: 2025-02-11 03:53:04 +0800 CST

Use purrr::accumulate com condição

  • 772

Abaixo está um exemplo reproduzível com um dataframe de teste:

dat<- structure(list(A = c(1.3, 1.5, 1.6, 1.2, 1.1, 1.2), 
                     B = c(0.25, 0.21, 0.21, 0.15, 0.26, 0.17)), 
                class = c("tbl_df", "tbl", "data.frame"), 
                row.names = c(NA, -6L))

Quero adicionar uma coluna com um valor inicial, digamos 1000, e então usar accumulate, mas dependendo de uma condição na coluna A.

Basicamente, se o valor na coluna A for superior ou igual a 1,2; Coluna C = 1000-(Coluna B * 1000). Mas isso é apenas para a primeira linha. Para o resto, em vez de 1000 (valor inicial), deve ser o valor anterior.

Minha saída desejada seria algo como isto:

UM B C
1.3 0,25 750
1.5 0,21 592,5
1.6 0,21 468.075
1.2 0,15 397.864
1.1 0,26 397.864
1.2 0,17 330.227

(e assim a primeira linha deve ser 1000 se a coluna A for inferior a 1,2).

Tentei usar accumulate, mas não consigo fazer a parte condicional funcionar:

dat<-dat %>%
  mutate(C = accumulate(tail(B,-1), .f = ~ .x - (.x * .y), .init = 1000))

2 respostas

  1. Best Answer

    Se você gostaria de usaraccumulate

    > dat %>%
+     mutate(C = accumulate(B * (A >= 1.2), ~ .x * (1 - .y), .init = 1000)[-1])
# A tibble: 6 × 3
      A     B     C
  <dbl> <dbl> <dbl>
1   1.3  0.25  750
2   1.5  0.21  592.
3   1.6  0.21  468.
4   1.2  0.15  398.
5   1.1  0.26  398.
6   1.2  0.17  330.

    Com a base R, você pode fazer o mesmo se usar Reduceoucumprod

    transform(
    dat,
    C = Reduce(\(x, y) x * (1 - y),
        B * (A >= 1.2),
        init = 1000,
        accumulate = TRUE
    )[-1]
)

    ou

    transform(
    dat,
    C = cumprod(1 - B * (A >= 1.2)) * 1000
)

    o que dá

        A    B        C
1 1.3 0.25 750.0000
2 1.5 0.21 592.5000
3 1.6 0.21 468.0750
4 1.2 0.15 397.8638
5 1.1 0.26 397.8638
6 1.2 0.17 330.2269

    dados

    > dput(dat)
structure(list(A = c(1.3, 1.5, 1.6, 1.2, 1.1, 1.2), B = c(0.25, 
0.21, 0.21, 0.15, 0.26, 0.17)), row.names = c(NA, -6L), class = c("tbl_df",
"tbl", "data.frame"))
    • 11 
  2. library(dplyr)

dat %>%
  mutate(C = purrr::accumulate2(B[-1], A[-1], ~ .x - (.x * .y) * (..3 >= 1.2) , 
                                .init = 1000 - first(B) * 1000 * (first(A) >= 1.2)))

#> # A tibble: 6 × 3
#>       A     B     C
#>   <dbl> <dbl> <dbl>
#> 1   1.3  0.25  750 
#> 2   1.5  0.21  592.
#> 3   1.6  0.21  468.
#> 4   1.2  0.15  398.
#> 5   1.1  0.26  398.
#> 6   1.2  0.17  330.

    Criado em 2025-02-10 com reprex v2.1.1

    • 7

relate perguntas