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

发布于03月08日

我有以下数据框，

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

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

我试着使用双花括号({{}})和:=，如下所示，

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

但我得到了以下错误，

然后我试着用下面的across()中的cur_column()来访问专栏，

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

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

推荐答案

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

大多数情况下，在每行的不同列上执行不同的选项并不容易.另一种方法是reshape 数据，以便使用超前/滞后函数.然后你可以转回来.

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

另一种 Select 是创建一个助手函数，该函数可以通过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)))