построение линии регрессии с решеткой - PullRequest
3 голосов
/ 13 января 2012

У меня тут небольшая проблема, пожалуйста, помогите мне.У меня есть эти данные

set.seed(4)
mydata <- data.frame(var = rnorm(100),
                     temp = rnorm(100),
                     subj = as.factor(rep(c(1:10),5)),
                     trt = rep(c("A","B"), 50))

и эта модель, которая соответствует им

lm  <- lm(var ~ temp * subj, data = mydata)

Я хочу построить результаты с решеткой и подгонять линию регрессии, предсказанную с помощьюмоя модель, через них.Для этого я использую этот подход, обрисованный в общих чертах «Решетки решетки для энергопотребления» Д. Саркара

temp_rng <- range(mydata$temp, finite = TRUE)

grid <- expand.grid(temp = do.breaks(temp_rng, 30),
                    subj = unique(mydata$subj),
                    trt = unique(mydata$trt))

model <- cbind(grid, var = predict(lm, newdata = grid))

orig <- mydata[c("var","temp","subj","trt")]

combined <- make.groups(original = orig, model = model)


xyplot(var ~ temp | subj, 
       data = combined,
       groups = which,
       type = c("p", "l"),
       distribute.type = TRUE
       )

Пока все хорошо, но я также хочу назначить цвет заливки дляТочки данных для двух обработок trt=1 и trt=2.

Итак, я написал этот фрагмент кода, который отлично работает, но когда дело доходит до построения линии регрессии, кажется, что тип не распознается функцией панели ...

my.fill <- c("black", "grey")

plot <- with(combined,
        xyplot(var ~ temp | subj,
              data = combined,
              group = combined$which,
              type = c("p", "l"),
              distribute.type = TRUE,
              panel = function(x, y, ..., subscripts){
                     fill <- my.fill[combined$trt[subscripts]] 
                     panel.xyplot(x, y, pch = 21, fill = my.fill, col = "black")
                     },
             key = list(space = "right",
                     text = list(c("trt1", "trt2"), cex = 0.8),
                     points = list(pch = c(21), fill = c("black", "grey")),
                     rep = FALSE)
                     )
      )
plot

Я также пытался переместить тип и распространять тип в пределах panel.xyplot, а также подмножество данных в нем panel.xyplot, например,

plot <- with(combined,
        xyplot(var ~ temp | subj,
              data = combined,
              panel = function(x, y, ..., subscripts){
                     fill <- my.fill[combined$trt[subscripts]] 
                     panel.xyplot(x[combined$which=="original"], y[combined$which=="original"], pch = 21, fill = my.fill, col = "black")
                     panel.xyplot(x[combined$which=="model"], y[combined$which=="model"], type = "l", col = "black")
                     },
             key = list(space = "right",
                     text = list(c("trt1", "trt2"), cex = 0.8),
                     points = list(pch = c(21), fill = c("black", "grey")),
                     rep = FALSE)
                     )
      )
plot

, но также безуспешно.

Может ли кто-нибудь помочь мне построить прогнозные значения в виде линий, а не точек?

Ответы [ 3 ]

6 голосов
/ 13 января 2012

Это может быть работа для пакета latticeExtra.

library(latticeExtra)
p1 <- xyplot(var ~ temp | subj, data=orig, panel=function(..., subscripts) {
  fill <- my.fill[combined$trt[subscripts]] 
  panel.xyplot(..., pch=21, fill=my.fill, col="black")
})
p2 <- xyplot(var ~ temp | subj, data=model, type="l")
p1+p2

enter image description here

Я не уверен, что происходит в вашей первой попытке, но тот, который имеетиндексы не работают, потому что x и y являются подмножеством данных для subj, поэтому их подмножество с использованием вектора на основе combined не будет работать так, как вы думаете.Попробуйте вместо этого.

xyplot(var ~ temp | subj, groups=which, data = combined,
       panel = function(x, y, groups, subscripts){
         fill <- my.fill[combined$trt[subscripts]]
         g <- groups[subscripts]
         panel.points(x[g=="original"], y[g=="original"], pch = 21, 
                      fill = my.fill, col = "black")
         panel.lines(x[g=="model"], y[g=="model"], col = "black")
       },
       key = list(space = "right",
         text = list(c("trt1", "trt2"), cex = 0.8),
         points = list(pch = c(21), fill = c("black", "grey")),
         rep = FALSE)
       )
2 голосов
/ 31 января 2013

Это может быть тривиально, но вы можете попробовать:

xyplot(... , type=c("p","l","r"))

"p" добавляет точки, "l" соединяет их пунктирными линиями, "r" соответствует линейной модели ваших данных. type="r" только построит только линии регрессии без отображения точек данных.

2 голосов
/ 13 января 2012

Может быть проще просто использовать функцию panel.lmline только для ваших исходных данных:

xyplot(var ~ temp | subj,
        data = orig,
        panel = function(x,y,...,subscripts){
            fill <- my.fill[orig$trt[subscripts]]
            panel.xyplot(x, y, pch = 21, fill = my.fill,col = "black")
            panel.lmline(x,y,col = "salmon")
        },
        key = list(space = "right",
                     text = list(c("trt1", "trt2"), cex = 0.8),
                     points = list(pch = c(21), fill = c("black", "grey")),
                     rep = FALSE)
)

enter image description here

...