具有分组变量的R中的回归

Question

使用自变量Mean ， X ， Y通过以下方式预测数据帧DF的因变量Value ：

DF <- DF %>% 
    group_by(Country, Sex) %>%
    do({ 
        mod = lm(Value ~ Mean + X + Y, data = .) 
        A <- predict(mod, .)
        data.frame(., A)
    })

数据按Country和Sex分组。 因此，拟合公式可以表示为：

Value(Country, Sex) = a0(Country, Sex) + a1(Country, Sex) Mean + a2(Country, Sex) X + a3(Country, Sex) Y

但是，我想使用以下公式：

Value(Country, Sex) = a0(Country, Sex) + a1(Country, Sex) Mean + a2(Country) X + a3(Country) Y

其中a2和a3与Sex无关。 我该怎么做？

Answer 1

我认为按Country和Sex分组时您无法做到。 您可以按Country分组并添加与Sex互动：

DF <- DF %>% 
group_by(Country) %>%
do({ 
    mod = lm(Value ~ Sex + Mean*Sex + X + Y, data = .) 
    A <- predict(mod, .)
    data.frame(., A)
})

或一次性添加与“ Sex和Country互动来估算模型：

mod <- lm(Value ~ Sex*Country*Mean + Country*X + Country*Y
A <- predict(mod)