r - 使用上一行的计算值来计算当前

Question

我有这个数据框

> df

# rn a b c  d  e  f
# 1  1 2 NA NA NA NA
# 2  5 8 NA 4  5  6 
# 3  8 5 4  2  3  2 
# 4  4 2 5  5  6  2

我正在尝试创建一个基于这些条件的新列:

• 当列c到f都是NA时，那么b
• 只有c为NA时，返回b和f列中的最小值，即min(b,f) 或 pmin
• 当行中不存在NA时，返回b、e、f中的最小值+上次计算的值

期望的输出是:

> df

# rn a b c  d  e  f  g
# 1  1 2 NA NA NA NA 2
# 2  5 8 NA 4  5  6  6  ### ? [b=8, f=6; least value = 6] 
# 3  8 5 4  2  3  2  3  ### ? [b=5, e=3, 'f + previous calculated value' = 2+6=8; least value = 3]
# 4  4 2 5  5  6  2  2  ### ? [b=2, e=6, 'f + previous calculated value' = 2+3=5; least value = 2]

我试过这个但我不知道如何访问以前计算的值(使用 lag(g) 作为占位符):

df%>%
  mutate(g = case_when(
      is.na(c) & is.na(d) & is.na(e) & is.na(f) ~ b,
      is.na(c) & !is.na(d) & !is.na(e) & !is.na(f) ~ pmin(b,f),
      !is.na(c) & !is.na(d) & !is.na(e) & !is.na(f) ~  pmin(b,e, f+lag(g)),
      TRUE ~ NA)
   )

也许我没有以正确的方式思考这个问题。任何帮助是极大的赞赏!

Answer 1

在这种情况下，我会说使用一个简单的 for 循环。您不能使用 lag(g)，因为您还没有构建 g 列。

    res <- rep(0, nrow(df))
    for (i in 1:nrow(df)) {
      row <- df[i, ]
      if (is.na(row["c"]) && is.na(row["f"])) {
        res[i] <- row["b"]
      } else if (is.na(row["c"])) {
        res[i] <- min(row["b"], row["f"])
      } else if (!is.na(row["d"])) {
        res[i] <- min(row["b"], row["e"], row["f") + res[i-1])
      }
    }
    df$g <- res