首页 > 解决方案 > 回归循环

问题描述

Ciao,我有几列代表分数。我想估计每个 SCORE 是 STUDYTIME 函数的模型。所以我想运行与 SCORE 列一样多的模型,所有简单的模型都是 STUDYTIME 的函数。然后我想将 STUDYTIME 的系数存储在一个新列中,该列的行名等于 SCORE 列名。最后,我不确定如何对线性模型进行聚类,因为学生每个人都在数据中两次。

这是我的复制示例。这是我现在拥有的数据:

df <- data.frame(replicate(5, rnorm(10)))
df[1]<-c(1,1,2,2,3,3,4,4,5,5)
colnames(df) <- c('student','studytime', 'score1','score2','score3')

这是我对编码的尝试:

for (i in 1:nrow(df)) {
  dfx         <- df[,i]
  lm    <- lm(dfx[,3:5] ~ study_time)
  resdat[,i] = summary(lm)$coefficients[2]
}

标签: rloopsstore

解决方案


您可以使用 simplelapplysapplyfunction 来做到这一点。

这是r代码:

生成数据

df <- data.frame(replicate(5, rnorm(10)))
df[1]<-c(1,1,2,2,3,3,4,4,5,5)
colnames(df) <- c('student','studytime', 'score1','score2','score3')

存储结果

Results <- lapply(df[, -c(1,2)], FUN = function(x) lm(x ~ df$studytime))
Coef <- sapply(Results, FUN = coefficients)

推荐阅读