dplyr中带有条件的递归函数答案

【问题标题】：Recursive function with condition in dplyrdplyr中带有条件的递归函数
【发布时间】：2018-02-19 17:53:17
【问题描述】：

首先是样本数据

        set.seed(123) 
        dat <- tibble(x = sample(-10:10, size = 11,replace = T))

        # A tibble: 11 x 1
              x
            <int>
        1    -4
        2     6
        3    -2
        4     8
        5     9
        6   -10
        7     1
        8     8
        9     1
        10    -1
        11    10

我有一个初始值为 2 的变量 y。我想通过在给定时间步中将 x 添加到 y 来计算 y 的最终值：

      y[i] = y[i-1] + x[i]    

     dat %>% mutate(y = accumulate(x, ~ .x + .y, .init = 2)[-1])

            # A tibble: 11 x 2
                  x     y
                <int> <dbl>
            1    -4    -2
            2     6     4
            3    -2     2
            4     8    10
            5     9    19
            6   -10     9
            7     1    10
            8     8    18
            9     1    19
            10    -1    18
            11    10    28

但是，我要强加的条件是，在给定的时间步长内，y 不能 > 10 或为负数。如果 y > 10, 10 并且如果 y

          # A tibble: 11 x 2
                  x     y    y1
                  <int> <dbl>
            1    -4    -2      0 (-2 converted to 0)
            2     6     4      6 
            3    -2     2      4
            4     8    10     10 (12 converted to 10)
            5     9    19     10 (19 converted to 10)
            6   -10     9      0  
            7     1    10      1
            8     8    18      9  
            9     1    19     10
            10    -1    18     9
            11    10    28    10  (19 converte to 10)

【问题讨论】：

老兄。您已将列命名为 x 和 y，此时，accumulate 使用几乎相同的 .x 和 .y 作为参数。这是一个非常令人困惑的例子。

标签： r dplyr purrr

【解决方案1】：

你只需要重新定义你用于accumulate的函数：

library(tidyverse)
set.seed(123) 
dat <- tibble(x = sample(-10:10, size = 11,replace = T))
fn <- function(x, y) pmax(pmin(x + y, 10), 0)
dat %>% 
  mutate(y = accumulate(x, fn, .init = 2)[-1])
dat
# A tibble: 11 × 2
       x     y
   <int> <dbl>
1     -4     0
2      6     6
3     -2     4
4      8    10
5      9    10
6    -10     0
7      1     1
8      8     9
9      1    10
10    -1     9
11    10    10

它是如何工作的：pmax 取两个值中的最大值，pmin 取最小值，因此您将 x+y 的总和包装到上限和下限中，以将结果限制在您需要的范围内。

【讨论】：

【解决方案2】：

kgolyaev 用非常相似的方法击败了我几分钟。

dat %>% mutate(y = accumulate(x, ~min(max(.x + .y, 0), 10), .init = 2)[-1])

【讨论】：

非常感谢。尽管如此，它还是清晰简洁的。