使用循环按组进行线性回归

Question

我有一个包含很多变量的数据集，我需要按国家进行线性回归。 我在这里有7个国家。 1个

现在，我想使用循环分别进行回归。 这是我的代码：

lmcountry(witd_hw$countryest,witd_hw$AverageIncome,witd_hw$sqincome)

虽然，似乎有一个错误：类型为'closure'的对象不可子集化

但实际上我可以像这样进行回归：

lm(witd_hw$countryest~witd_hw$AverageIncome+witd_hw$sqincome,data=witd_hw[witd_hw$Country=="China",])

Answer 1

不是通过for循环，而是通过list apply函数。 首先按国家将数据框划分为子框。 然后将lm应用于每个子集：

# Reproducible data example
df = data.frame(x = rnorm(100), y = rnorm(100), country = sample(c('A', 'B', 'C'), 100, replace=TRUE))

list_of_country_dfs = split(df, df$country)

results = lapply(list_of_country_dfs, function(dat) lm(y ~ x, data = dat))
lapply(results, summary)