使用 R 调用多个函数的优雅方式

Question

我有一组预处理函数，我最终调用它们来创建最终数据帧。如您所见，每个阶段的输出都作为输入传递给下一个阶段。

wrap_func = function(DF){  # Wrapper function which has all the above functions put together. 
T1 = transform_ids(DF)                                                          
print("Id transformation success")   #print statements are used for debugging purposes
T2 = transform_dates(T1)
print("date transformation success")
T3 = measurement_ids(T2)
print("measurement_ids transformation success")
T4 = oper_val_concepts(T3)
print("operator transformation success")
T5 = range_values(T4)
print("range capture success")
T6 = col_format(T5)
print("column formatting success")
T7 = missing_impute(T6,def_val)
print("Missing value Imputation success")
T7 = T7[c(                                                     # reordering the column display
"measurement_id","person_id")]
 return(T7)
 } 

DF = wrap_func(dfm)

有什么优雅的写法吗？

这篇文章有类似的场景，但它是在 Python 中。related post

你能帮我用 R 把它变得优雅吗？

Answer 1

一种解决方案是：

pipeline <- function(DF){
 DF %>%
 transform_ids() %T>%                                                          
  cat("Id transformation success\n") %>%
 transform_dates() %T>%
  cat("date transformation success\n") %>%
 measurement_ids() %T>%
  cat("measurement_ids transformation success\n") %>%
 oper_val_concepts() %T>%
  cat("operator transformation success\n") %>%
 range_values() %T>%
  cat("range capture success\n") %>%
 col_format() %T>%
  cat("column formatting success\n") %>%
 missing_impute(def_val) %T>%
  cat("Missing value Imputation success\n") %>%
 .[c("measurement_id","person_id")]
} 

DF <- pipeline(dfm)

地点：

• magrittr 的%>% 允许您将左侧的结果作为第一个参数传递给下一个函数。

• magrittr 的%T>% 允许您返回左侧值（因为您不想将打印的字符串通过管道传递到下一步）

• 末尾的点 . 指的是管道对象（这里是数据框）

如果您更喜欢使用print() 而不是cat，请务必将其放在大括号{print("Hello World!"} 之间

如果您愿意放弃调试消息（或将其集成到每个独特的功能中）您可以使用purrr的compose：

pipeline <- compose(~ .x[c("measurement_id","person_id")], ~ missing_impute(.x, def_val), col_format, range_values, oper_val_concepts, measurement_ids, transform_dates, transform_ids)

DF <- pipeline(dfm)

请注意，这里的函数是从右到左应用的（但您可以使用选项 compose(..., .dir = "forward") 从左到右应用）