簡體   English   中英

將data.frame列名傳遞給函數

[英]passing data.frame column names to a function

我見過類似的文章,其中提到需要使用引號將列名傳遞給函數,但是我可以在遇到問題以及如何改進函數時使用幫助。 例如,也許我可以添加“后綴”參數以自動為新的data.frame分配帶有原始后綴的名稱? 我希望能夠有一個通用函數,可用於更改列名和位置。 謝謝。

library(tidyverse)  

# function definition #
  createhrly_0595quants <- function(df, hourcolumn, 
     value, qtype, metadata_to_add)   {

df <- df  %>% group_by(hourcolumn) %>%
         summarize(`05%`=quantile(value, probs=0.05, type =qtype),
        `95%`=quantile(value, probs=0.95, type = qtype),
         median = median(value), n=n()) %>% 
         mutate(qtype = qtype, metadata_to_add = metadata_to_add)

}

# sample data.frame #
hrly_gmt  <- seq(from=as.POSIXct("2018-11-20 01:00", 
tz="America/Los_Angeles"), to=as.POSIXct("2018-11-20 23:00", 
 tz="America/Los_Angeles"), by="1 hours")  
myvalues1 <- rnorm(23)
myvalues2 <- rnorm(23)
mydf1 <- data.frame(hrly_gmt, myvalues1) %>% mutate(class = "a")
mydf2 <- data.frame(hrly_gmt, myvalues2) %>% mutate(class = "b")
df_x <- rbind(mydf1, mydf2)

# function use #
df_0595quants <- createhrly_0595quants(df_x, "hrly_gmt",
                 "myvalues", 4, "version x.2")

當我們將字符串作為輸入傳遞時,而不是使用group_by ,我們可以使用group_by_at ,它使用字符串作為列名, summarize列可以轉換為符號( sym )並求值( !!

createhrly_0595quants <- function(df, hourcolumn, 
 value, qtype, metadata_to_add)   {

  value <- rlang::sym(value)


    df  %>% 
      group_by_at(vars(hourcolumn)) %>%
      summarize(`05%`=quantile(!!value, probs=0.05, type =qtype),
              `95%`=quantile(!!value, probs=0.95, type = qtype),
             median = median(!!value), n=n()) %>% 
     mutate(qtype = qtype, 
     metadata_to_add = metadata_to_add)
 }



createhrly_0595quants(df_x, "hrly_gmt",
                  "myvalues", 4, "version x.2")

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM