一个参数可以映射到用户定义函数中的多个参数吗?

时间:2017-03-23 08:36:21

标签: r function dplyr user-defined-functions

假设我想运行它:

MS_date<-bind_inpatient_MSW %>% 
  arrange(NRIC,
          APPROVED_DATE_BILL,APPROVED_DATE_FF_APPLICATION) %>%
  group_by(NRIC,
           APPROVED_DATE_BILL,APPROVED_DATE_FF_APPLICATION) %>%
  mutate(n_marital_status=n_distinct(MARITAL_STATUS,na.rm=TRUE))

和这个

TH_date<-bind_inpatient_MSW %>% 
  arrange(NRIC,
          APPROVED_DATE_BILL) %>%
  group_by(NRIC,
           APPROVED_DATE_BILL) %>%
  mutate(n_TH=n_distinct(TYPE_OF_HOUSING,na.rm=TRUE))

这两者的不同之处在于对数据框进行排列和分组的变量,以及添加的变量。我想写一个用户定义的函数,这样我就不必多次写这个了。我尝试如下:

df_date<-function(df,grpby,cntby){
  dfnew<-df %>%
    arrange(grpby) %>%
    group_by(grpby) %>%
    mutate(n=n_distinct(cntby,na.rm=TRUE))
  return(dfnew)
}

并应用df_date(bind_inpatient_MSW,NRIC,APPROVED_DATE_BILL,APPROVED_DATE_FF_APPLICATION,MARITAL_STATUS)

df_date(bind_inpatient_MSW,NRIC,APPROVED_DATE_BILL,TYPE_OF_HOUSING)

他们不会工作。我怎么能解决这个问题?

1 个答案:

答案 0 :(得分:0)

您可以尝试以下内容:

fun <- function(dat,group,ctnby) {
            dat %>% 
            group_by_(group) %>%
            do((function(., ctnby) {
                with(., data.frame(n = n_distinct(get(ctnby))))
            } 
    )(.,ctnby))
    }

fun(mtcars,"cyl","hp")

使用do避免了延迟评估。