时间戳

首页 > 解决方案 > 将 data.frame 列名传递给函数

r - 将 data.frame 列名传递给函数

问题描述

我看过类似的帖子，提到需要使用引号将列名传递给函数，但我可以使用帮助来解决我的错误以及如何改进函数。例如，也许我可以添加一个“后缀”参数来自动为新分配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")

标签： rtidyverse

解决方案

group_by当我们将字符串作为输入传递时，我们可以使用group_by_at将字符串作为列名而不是 using ，该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")

推荐阅读