在變量中使用變量參數mutate_at

Question

我想寫一個簡單的函數，它可以使用通過調用mutate_at中的省略號傳遞的變量。

# Packages 
sapply(
    X = c("dplyr", "rlang"), 
    FUN = require, 
    character.only = TRUE 
) 

# Data 
set.seed(1) 
cbind(as.data.frame(replicate(3, replicate(
    10, paste0(sample(letters, 5), collapse = "") 
))), c(1:10)) %>% setNames(nm = paste(replicate(4, "col"), LETTERS[1:4], sep = "_")) -> dta 


# Mutate 
mutate_some <- function(df, ...) { 
    mut_vars <- quos(...) 
    df %>% 
     mutate(.vars = !!!mut_vars, 
       .funs = funs(toupper(.))) 
} 

# test 
dta %>% mutate_some(col_A, col_C) 

問題

的代碼產生：

>> dta %>% mutate_some(col_A, col_C) 
    col_A col_B col_C col_D .vars1 .vars2  .funs 
1 gjnue mvkfb rigwn  1 gjnue rigwn ~toupper(.) 
2 xzpob chmpi fdlvn  2 xzpob fdlvn ~toupper(.) 
3 feqix xhlyo zsijd  3 feqix zsijd ~toupper(.) 
4 mrxiy glsbt arcko  4 mrxiy arcko ~toupper(.) 
5 yfpcz iuzhk zmldq  5 yfpcz zmldq ~toupper(.) 
6 kajuh xvjry lmefn  6 kajuh lmefn ~toupper(.) 
7 moles lrjhq obazu  7 moles obazu ~toupper(.) 
8 rtcqj frczd pomwl  8 rtcqj pomwl ~toupper(.) 
9 vqsml gbpur rpfxq  9 vqsml rpfxq ~toupper(.) 
10 ualqp uljsn lerct 10 ualqp lerct ~toupper(.) 

期望的結果

結果應相當於：

dta %>% mutate_at(.funs = funs(toupper(.)), .vars = c("col_A", "col_C")) 
    col_A col_B col_C col_D 
1 GJNUE mvkfb RIGWN  1 
2 XZPOB chmpi FDLVN  2 
3 FEQIX xhlyo ZSIJD  3 
4 MRXIY glsbt ARCKO  4 
5 YFPCZ iuzhk ZMLDQ  5 
6 KAJUH xvjry LMEFN  6 
7 MOLES lrjhq OBAZU  7 
8 RTCQJ frczd POMWL  8 
9 VQSML gbpur RPFXQ  9 
10 UALQP uljsn LERCT 10 

---

個

假設：

• 的mutate_some函數必須採取省略參數是在使用mut_vars <- quos(...)鱸。

• 特別是，我正在尋找一個類似於rlang::do_something_(mut_vars)的解決方案。此功能是一個較長的管道的一部分，我需要保留mut_vars爲quos()，因爲我稍後使用它。

• 我不想更改列名（如上面所需的結果）。

Answer 1

你似乎已經意識到mutate_at功能的，我不知道爲什麼你是不是在你的mutate_some使用它，但它會產生你想要

mutate_some <- function(df, ...) { 
    mut_vars <- quos(...) 
    df %>% 
    mutate_at(mut_vars, 
      .funs = funs(toupper(.))) 
}