如何使用 tidyverse 中的相邻列修改在另一列中命名的列？

Question

我有以下数据框，

df <- tibble(x = c(2, 3, 4)) %>% 
mutate(`1` = 99, `2` = 88, `3` = 77, `4` = 66, `5` = 55)

列 x 包含需要操作的列名称，该列中的值必须替换为列 x-1x-1 中的值之和代码>、<代码>x和<代码>x+1。例如，对于 x 为 2 的第一行，2 列中的值必须替换为 (99+88+77) = 264。

我尝试使用双卷曲括号({{}}) 和 :=，如下所示，

df %>% 
  mutate("{{x}}" := {{x-1}} + {{x}} + {{x+1}})

但出现以下错误， local_error_context(dots =dots, .index = i, mask = mask) 中的错误：promise 已在评估中：递归默认参数引用或早期问题？

然后我尝试使用 cur_column( ) 在 across() 内，如下所示，

df %>% 
  mutate(across(-x, ~if_else(x == cur_column(), {{cur_column()}}, .x)))

我得到了与上面相同的错误，我想我可能错误地使用了卷曲运算符，有人可以帮忙吗？

Answer 1

{{}} 语法适用于将未计算的表达式传递给 dplyr 命令时，它不适用于捕获列值。

大多数时候，在每行的不同列上执行不同的选项并不容易。一种替代方法是重塑数据，以便可以使用超前/滞后函数。然后你就可以向后旋转。

library(dplyr)
library(tidyr)
df %>% 
  mutate(row = row_number()) %>% 
  pivot_longer(!c(x, row)) %>% 
  mutate(name = as.integer(name)) %>% 
  group_by(row) %>% 
  mutate(value=if_else(x==name, value + lead(value) + lag(value), value)) %>% 
  pivot_wider(c(row, x)) %>% 
  select(-row)

#     row     x   `1`   `2`   `3`   `4`   `5`
#   <int> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
# 1     1     2    99   264    77    66    55
# 2     2     3    99    88   231    66    55
# 3     3     4    99    88    77   198    55

另一种选择是创建一个辅助函数，可以使用 rowwise() 访问 cur_column() 和 cur_data()，以便创建不同的每行的转换。

colclump <- function(target) {
  prevc <- as.character(as.integer(target)-1)
  nextc <- as.character(as.integer(target)+1)
  function(x) {
    if (cur_column()==target) {
      x + cur_data()[[prevc]] + cur_data()[[nextc]]
    } else {
      x
    }
  }
  
}
df %>% 
  rowwise() %>% 
  mutate(across(-x, ~colclump(x)(.x)))

Answer 2

也许这也符合你的目的：

for(k in 2:4) {
  df[k-1,k+1] <- df[k-1,as.character(df$x[k-1])] + 
              df[k-1,as.character(df$x[k-1]-1)] + 
              df[k-1,as.character(df$x[k-1]+1)]
}
df
#  A tibble: 3 x 6
# x   `1`   `2`   `3`   `4`   `5`
# <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
# 1     2    99   264    77    66    55
# 2     3    99    88   231    66    55
# 3     4    99    88    77   198    55