聚合线性回归

Question

对不起，我对R非常陌生，但我有一个数据与多个玩家的游戏。我试图得到的斜率系数，每个球员的积分在他们的所有游戏。我已经看到aggregate可以使用像sum和average这样的运算符，从线性回归中得到系数也很简单。我怎么把它们结合起来呢？

a <- c("player1","player1","player1","player2","player2","player2")
b <- c(1,2,3,4,5,6)
c <- c(15,12,13,4,15,9)
gamelogs <- data.frame(name=a, game=b, pts=c)

我想让这个变成：

   name    pts slope
player1       -.4286
player2       .08242    

Answer 1

您还可以使用基本lm进行一些魔术，一次完成所有操作：

coef(lm(game ~ pts*name - pts, data=gamelogs))[3:4]
coef(lm(game ~ pts:name + name, data=gamelogs))[3:4]
#pts:nameplayer1 pts:nameplayer2 
#    -0.42857143      0.08241758 

作为一个data.frame

data.frame(slope=coef(lm(game ~ pts*name - pts, data=gamelogs))[3:4])
#                      slope
#pts:nameplayer1 -0.42857143
#pts:nameplayer2  0.08241758

有关lm调用中的建模的进一步解释，请参见此处：

https://stat.ethz.ch/R-manual/R-devel/library/stats/html/formula.html

http://faculty.chicagobooth.edu/richard.hahn/teaching/FormulaNotation.pdf#2

在本例中，pts*name扩展为pts + name + pts:name，当移除- pts时，这意味着它等同于pts:name + name。

Answer 2

你可以

s <- split(gamelogs, gamelogs$name)

vapply(s, function(x) lm(game ~ pts, x)[[1]][2], 1)
#     player1     player2 
# -0.42857143  0.08241758 

或

do.call(rbind, lapply(s, function(x) coef(lm(game ~ pts, x))[2]))
#                 pts
# player1 -0.42857143
# player2  0.08241758

或者如果你想使用dplyr，你可以

library(dplyr)

models <- group_by(gamelogs, name) %>% 
    do(mod = lm(game ~ pts, data = .))

cbind(
    name = models$name, 
    do(models, data.frame(slope = coef(.$mod)[2]))
)
#      name       slope
# 1 player1 -0.42857143
# 2 player2  0.08241758

Answer 3

库nlme也有此功能，lmList

library(nlme)
coef(lmList(game ~ pts | name, gamelogs))
#        (Intercept)         pts
# player1    7.714286 -0.42857143
# player2    4.230769  0.08241758