具有分組變量的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)