如何构建从data.table到magrittr并返回data.table的管道

时间:2015-03-07 22:11:23

标签: r data.table magrittr

我想将data.table管道与magrittr流水线混合。我可以从data.table转到%>%但我无法弄清楚如何回到[] [] data.table样式的流水线。

以下是一个例子:

> tbl = data.table(grp=c(1,1,1,2,2,2,3,3,3,4,4), y=rnorm(11))
> tbl
    grp        y
 1:   1  0.08150
 2:   1  1.51330
 3:   1 -0.26154
 4:   2 -0.12746
 5:   2  0.10747
 6:   2  0.16502
 7:   3  0.54139
 8:   3 -0.04194
 9:   3  0.02373
10:   4  2.00756
11:   4  1.05523
> tbl[, .(.N, mean(y)), by=grp][order(-N)] %>% head(n=3) %>% .[, N := NULL]
   grp      V2
1:   1 0.44442
2:   2 0.04834
3:   3 0.17439
> tbl[, .(.N, mean(y)), by=grp][order(-N)] %>% head(n=3) %>% .[, N := NULL][, plot(grp, V2)]
Error in `[.data.table`(., .[, `:=`(N, NULL)], , plot(grp, V2)) : 
  'by' or 'keyby' is supplied but not j
Calls: %>% ... freduce -> withVisible -> <Anonymous> -> [ -> [.data.table
> 

如何在%&gt;%之后返回[] []?

我知道这个特定的例子可以完全用[]和没有%&gt;%重写,但我不想每次都这样做。我想要一种能够写出[] []%&gt;%[] []模式的方法。

3 个答案:

答案 0 :(得分:2)

你可以做到

 `tbl %>% filter(y>0) %>% data.table()` 

将管道结果转换为data.table,例如,以data.table方式很好地打印结果。但是,不幸的是,你不能做像

这样的事情
 `tbl %>% filter(y>0) %>% data.table() [, mean(y), by=group]

我想知道,如果可以将此功能添加到将来的data.table版本中 - 可能通过新语法(以克服优先顺序限制,因为“[”在“&lt;”之前执行)。

答案 1 :(得分:2)

之前的答案都忽略了您在某种程度上指定优先顺序的能力。您可以通过将代码的%>%部分的优先级括在{} s:

中来实现此目的。
x <- data.frame(a=1:5, b=6:10)
{x %>% subset(a<4) %>% data.table()}[, mean(b)]

不漂亮,但确实有效:

> {x %>% subset(a<4) %>% data.table()} [, mean(b)]
[1] 7

答案 2 :(得分:1)

你做不到。 [higher precedence而不是%any%,所以我们会先评估它。