将不同的数字添加到dataframe的不同列

Question

以下是我目前的尝试：

library("tidyverse")
library("rlang")
data(mtcars)

mtcars_colnames <- quo(c("mpg", "cyl", "disp"))
num <- c(1, 2, 3)

mtcars %>% mutate(across(!!! mtcars_colnames, function(x) {x + num[col(.)]}))

我一直在思考如何动态地向列(mpg、cyl、disp)添加(1，2，3)。提前谢了。

Answer 1

我们可以通过只传递字符串向量而不是quo保证和'num'的命名向量来更改输入，然后使用跨中的cur_column与'num'的命名向量匹配，获得相应的值并进行加法

library(dplyr)
mtcars_colnames <- c("mpg", "cyl", "disp")
num <- setNames(c(1, 2, 3), mtcars_colnames)
mtcars1 <- mtcars %>%
    mutate(across(all_of(mtcars_colnames), ~ num[cur_column()] + .))

-检查输出

# // old data
mtcars %>% 
    select(all_of(mtcars_colnames)) %>% 
    slice_head(n = 5)
#                   mpg cyl disp
#Mazda RX4         21.0   6  160
#Mazda RX4 Wag     21.0   6  160
#Datsun 710        22.8   4  108
#Hornet 4 Drive    21.4   6  258
#Hornet Sportabout 18.7   8  360

# // new data
mtcars1 %>% 
    select(all_of(mtcars_colnames)) %>%
    slice_head(n = 5)
#                   mpg cyl disp
#Mazda RX4         22.0   8  163
#Mazda RX4 Wag     22.0   8  163
#Datsun 710        23.8   6  111
#Hornet 4 Drive    22.4   8  261
#Hornet Sportabout 19.7  10  363

或者，如果我们希望传递一个未命名的'num'向量，那么cur_column将curl_column与across`中的'mtcars_colnames匹配以返回索引，然后使用该索引来子集'num'

mtcars1 <- mtcars %>%
    mutate(across(all_of(mtcars_colnames), 
           ~ num[match(cur_column(), mtcars_colnames)] + .))