根据 R 中的先前值填写列

Question

我正在寻找一个整洁的解决以下问题。

我有以下示例数据集：

ex_tbl <- tibble(col1 = c(5, NA, NA, NA), col2 = c(1, 1.3, 0.8, 1.2))

我需要根据列2和列1中的前一个值填写NAs。如果我从第一个NA(第1列，第2行)开始，并且第2列大于1，那么我将把第1列的值(第1行)乘以第2列的值。那么如果col2小于等于1，那么col1正好等于前面的值。

这应该是最终结果：

  col1  col2
     5   1  
    6.5  1.3
    6.5  0.8
    7.8  1.2

我尝试了case_when（）但没有得到想要的结果：

output <- ex_tbl %>% mutate(col1 = case_when(col2 > 1 ~ col1 * col2,
                                              col2 <= 1 ~ col1,
                                              TRUE ~ col1)

显然，我没有捕获我想要使用前一行的值的部分。任何帮助都将不胜感激。

Answer 1

我们可能需要在这里累积

library(dplyr)
library(purrr)
ex_tbl %>% 
   mutate(coln = accumulate(col2, ~  if(.y <= 1) .x else .x * .y , 
         .init = first(col1))[-1])

-输出

# A tibble: 4 × 3
   col1  col2  coln
  <dbl> <dbl> <dbl>
1     5   1     5  
2    NA   1.3   6.5
3    NA   0.8   6.5
4    NA   1.2   7.8

Answer 2

也许你可以用

library(dplyr)

ex_tbl %>% 
  mutate(col1 = first(col1) * cumprod(pmax(col2, 1)))

要得到

# A tibble: 4 x 2
   col1  col2
  <dbl> <dbl>
1   5     1  
2   6.5   1.3
3   6.5   0.8
4   7.8   1.2

如果您想在< code>col2的情况下乘以第三列的值

ex_tbl <- tibble(col1 = c(5, NA, NA, NA), col2 = c(1, 1.3, 0.8, 1.2), col3 = c(0.5, 2, 2, 0.3))

ex_tbl %>% 
  mutate(
    factor = (col2 >= 1) * col2 + (col2 < 1) * col3,
    col1 = first(col1) * cumprod((col2 >= 1) * col2 + (col2 < 1) * col3))

注意：这里的factor列不是必需的。它只是显示cumprod函数内部发生的事情。