创建回归线上方数据点的数据表

Question

我有一个适合固定效应模型的面板数据集：

data <- data.frame(ID = c(1,1,1,1,2,2,3,3,3),
                   year = c(1,2,3,4,1,2,1,2,3),
                   progenyMean = c(90,78,92,69,86,73,82,85,91),
                   damMean = c(89,89,72,98,95,92,94,87,89)

ID, year, progenyMean, damMean
1, 1, 70, 69
1, 2, 68, 69
1, 3, 72, 72
1, 4, 69, 68
2, 1, 76, 75
2, 2, 73, 80
3, 1, 72, 74
3, 2, 75, 67
3, 3, 71, 69

# Fixed Effects Model in plm
fixed <- plm(progenyMean ~ damMean, data, model= "within", index = c("ID","year"))

我用蓝色的固定效应回归线绘制了 progenyMean 与 damMean 的关系图：

我想确定固定效应回归线上方的ID并创建此信息的数据表。 每个唯一ID都有多个数据点，因此一个ID可能在回归线的上方和下方都有点。

我已经计算了固定效应模型（ fitted ）的预测值并确定了该线上方的数据点：

fitted <- as.numeric(fixed$model[[1]] - fixed$residuals)
above <- data$progenyMean3Y > fitted
above
 [1]  TRUE FALSE  TRUE FALSE  TRUE FALSE  TRUE FALSE FALSE  TRUE FALSE
[12] FALSE FALSE FALSE  TRUE FALSE  TRUE FALSE FALSE  TRUE FALSE  TRUE
[23] FALSE  TRUE  TRUE FALSE FALSE  TRUE  TRUE FALSE FALSE  TRUE FALSE

是否可以使用data.table创建一个表，其中每一行都是一个唯一的ID并且有两列， above和below计算该ID的回归线上方/下方的数据点数？

Answer 1

按照您在问题中已经提出的逻辑，以下是构建摘要data.table一种方法：

library(data.table)
library(plm)

data <- data.frame(ID = c(1,1,1,1,2,2,3,3,3),
                   year = c(1,2,3,4,1,2,1,2,3),
                   progenyMean = c(90,78,92,69,86,73,82,85,91),
                   damMean = c(89,89,72,98,95,92,94,87,89))

# Fixed Effects Model in plm
fixed <- plm(progenyMean ~ damMean, data, model= "within", index = c("ID","year"))

fitted <- as.numeric(fixed$model[[1]] - fixed$residuals)
above <- data$progenyMean > fitted

dt <- as.data.table(data)

dt[, `:=`(fitted = fixed$model[[1]] - fixed$residuals,
          above = progenyMean > fitted,
          below = progenyMean < fitted)
   ][,
     .(above = sum(above), below = sum(below)), by = "ID"][]

#>    ID above below
#> 1:  1     1     3
#> 2:  2     1     1
#> 3:  3     1     2

^{由reprex 包(v2.0.1) 于 2021 年 8 月 27 日创建}