Question

我有一个data.table这样的对象

library(data.table)

a <- structure(list(PERMNO = c(10006L, 10006L, 10015L, 10015L, 20000L, 20000L), 
                    SHROUT = c(1427L, 1427L, 1000L, 1001L, 200L, 200L), 
                    PRC = c(6.5, 6.125, 0.75, 0.5, 3, 4), 
                    RET = c(0.005, -0.005, -0.001, 0.05, -0.002, 0.0031)),
                   .Names = c("PERMNO", "SHROUT", "PRC", "RET"), 
               class = c("data.table", "data.frame"), row.names = c(NA, -6L))

setkey(a,PERMNO)

我需要通过PERMNO执行一些计算，但是在这个示例中，假设它们只是2：

mktcap <- a[ , tail(SHROUT,n=1)*tail(PRC,n=1),by=PERMNO]
sqret <- a[, sum(RET^2),by=PERMNO]

产生

> mktcap
     PERMNO       V1
[1,]  10006 8740.375
[2,]  10015  500.500
[3,]  20000  800.000

> sqret
     PERMNO        V1
[1,]  10006 5.000e-05
[2,]  10015 2.501e-03
[3,]  20000 1.361e-05

我想将两个函数合并为一个，以产生一个包含3列的矩阵（或data.table，data.frame，等等），第一列为PERMNO，第二列为{{ 1}}和第三个mktcap。

问题是这个分组功能（即sqrt）似乎只产生两列的结果，一列有键，另一列有结果。

这是我尝试产生我想要的东西之一（

）

variable[ , function(), by= ]

产生

comb.fun <- function(datai) {
     mktcap <- as.matrix(tail(datai[,1],n=1)*tail(datai[,2],n=1),ncol=1)
     sqret <- as.matrix(sum(datai[,3]^2),ncol=1)
     return(c(mktcap,sqret))
}   

myresults <- a[, comb.fun(cbind(SHROUT,PRC,RET)), by=PERMNO]

（结果都在那里，但他们被迫进入一个专栏）。无论我尝试什么，我都无法分组返回一个包含两列以上（或多列结果）的矩阵。

是否可以通过PERMNO V1 [1,] 10006 8.740375e+03 [2,] 10006 5.000000e-05 [3,] 10015 5.005000e+02 [4,] 10015 2.501000e-03 [5,] 20000 8.000000e+02 [6,] 20000 1.361000e-05？

分组获得两列或更多列结果

Answer 1

答案（使用list()收集几个所需的摘要统计信息）在?data.table帮助文件的优秀示例部分中。（距离底部大约20行）。

out <- a[ , list(mktcap = tail(SHROUT,n=1)*tail(PRC,n=1),
                 sqret  = sum(RET^2)),
         by=PERMNO]

out
#    PERMNO   mktcap     sqret
# 1:  10006 8740.375 5.000e-05
# 2:  10015  500.500 2.501e-03
# 3:  20000  800.000 1.361e-05

修改

在下面的评论中，Matthew Dowle描述了一种清理代码的简单方法，其中j等调用中的x[i,j,by]参数变得笨拙。

在上面的电话中实施他的建议，你可以改为：

## 1) Use quote() to make an expression object out of the statement passed to j mm <- quote(list(mktcap = tail(SHROUT,n=1)*tail(PRC,n=1), sqret = sum(RET^2))) ## 2) Use eval() to evaluate it as if it had been typed directly in the call a[ , eval(mm), by=PERMNO] # PERMNO mktcap sqret # 1: 10006 8740.375 5.000e-05 # 2: 10015 500.500 2.501e-03 # 3: 20000 800.000 1.361e-05

Answer 2

怎么样

comb.fun <- function(a) {
 mktcap <- a[ , tail(SHROUT,n=1)*tail(PRC,n=1),by=PERMNO]
 sqret <- a[, sum(RET^2),by=PERMNO]

 return(merge(mktcap,sqret))
}

在data.table中分组：如何获得超过1列的结果？

2 个答案: