繁体   English   中英

使用reshape2 :: melt和reshape2 :: dcast在用户定义的函数中调用变量

[英]Calling variable in user-defined function with reshape2::melt and reshape2::dcast

我想转换这个数据框

data <- data.frame(color=c("red","red","red","green","green","green","blue","blue","blue"),object=c("box","chair","table","box","chair","table","box","chair","table"),units=c(1:9),price=c(11.5,12.5,13.5,14.5,15.5,16.5,17.5,18.5,19.5))

到另一个

output <- data.frame(color=c("red","green","blue"),units_box=c(1,4,7),price_box=c(11.5,14.5,17.5), units_chair=c(2,5,8),price_chair=c(12.5,15.5,18.5),units_table=c(3,6,9),price_table=c(13.5,16.5,19.5))

因此,我使用reshape2::meltreshape2::dcast来构建用户定义的函数,如下所示

fun<-function(df,var,group){
  r<-reshape2::melt(df,id.vars=var)
  r<-reshape2::dcast(r,var~group) 
return(r)
}

当我如下使用函数时

有趣的(数据,颜色,对象)

我收到以下错误消息

melt_check(数据,id.vars,measure.vars,variable.name,value.name)中的错误:找不到对象'颜色'

你知道我该怎么解决吗? 我认为问题是我应该在reshape2::melt中用引号调用变量,但我不知道如何。

注意1:我想保留变量的原始数字格式(即不带小数的对象和带小数的价格)

注2:我想指出,我的实际代码(这只是一个简化的示例)要长得多,并且涉及dplyr函数(包括enquo()和UQ()函数)。 因此,这种情况下的解决方案应与dplyr兼容。

注3:我不使用tidyr(我对整个tidyverse感到很开心),因为当前的tidyr仍然使用旧语言编写函数,并且与其他不愿意使用tidyr开发版本的人共享脚本。 。

我们可以使用dcastdata.table

library(data.table)
dcast(setDT(data), color ~object, value.var = c("units", "price"), FUN = c(length, mean))

我自己解决了这个问题(尽管我不太清楚背后的原因)。 我怀疑主要的问题是在melt和dcast中传递用户定义函数的变量,这可能是由于缺少引号(?)导致某种冲突。 无论如何,我使用dplyr :: rename重命名了变量,以便名称不再取决于变量,而取决于字符。 在这里,您可以看到我正在应用的最终代码:

fun<-function(df,var,group){
  enquo_var<-enquo(var)
  enquo_group<-enquo(group)

  r<-df%>%
    reshape2::melt(., id.var=1, variable.name = "parameter")%>%
    dplyr::rename(var = UQ(enquo_var))%>%
    reshape2::dcast(data=., formula = var~parameter, value.var = "value")

return(r)
}

funx<-fun(data,color,object)

尽管我找到了解决自己特定问题的方法,但如果有人向我解释其背后的原因,我将不胜感激。

PS:无论如何,我希望新版本的Tidyr可以使这些任务变得更容易。 感谢@hadley所做的出色工作。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM