Làm thế nào một âm mưu có thể liên tục bằng các tương tác liên tục trong ggplot2?

Hãy nói rằng tôi có dữ liệu:

x1 <- rnorm(100,2,10)
x2 <- rnorm(100,2,10)
y <- x1+x2+x1*x2+rnorm(100,1,2)
dat <- data.frame(y=y,x1=x1,x2=x2)
res <- lm(y~x1*x2,data=dat)
summary(res)

Tôi muốn vẽ biểu đồ liên tục bằng tương tác liên tục sao cho x1 nằm trên trục X và x2 được biểu thị bằng 3 dòng, một dòng biểu thị x2 ở điểm Z là 0, một ở điểm Z là +1 và khác ở điểm Z Điểm Z là -1, với mỗi dòng một màu riêng biệt và được dán nhãn. Làm thế nào tôi có thể làm điều này bằng cách sử dụng ggplot2?

Ví dụ: nó có thể trông giống như thế này (mặc dù tất nhiên với các dòng màu khác nhau thay vì các loại dòng khác nhau):

Đây là phiên bản của tôi với bộ dữ liệu mô phỏng của bạn:

x1 <- rnorm(100,2,10)
x2 <- rnorm(100,2,10)
y <- x1+x2+x1*x2+rnorm(100,1,2)
dat <- data.frame(y=y,x1=x1,x2=x2)
res <- lm(y~x1*x2,data=dat)
z1 <- z2 <- seq(-1,1)
newdf <- expand.grid(x1=z1,x2=z2)

library(ggplot2)
p <- ggplot(data=transform(newdf, yp=predict(res, newdf)), 
            aes(y=yp, x=x1, color=factor(x2))) + stat_smooth(method=lm)
p + scale_colour_discrete(name="x2") + 
  labs(x="x1", y="mean of resp") + 
  scale_x_continuous(breaks=seq(-1,1)) + theme_bw()

Tôi cho phép bạn quản lý các chi tiết về nhãn trục x / y và định vị chú giải.

Tính toán các ước tính cho y với điểm Z là 0 ( cột y0 ), -1 ( cột y1m ) và 1 ( cột y1p ):

dat$y0 <- res$coefficients[[1]] + res$coefficients[[2]]*dat$x1 + res$coefficients[[3]]*0 + res$coefficients[[4]]*dat$x1*0
	dat$y1m <- res$coefficients[[1]] + res$coefficients[[2]]*dat$x1 + res$coefficients[[3]]*-1 + res$coefficients[[4]]*dat$x1*-1
dat$y1p <- res$coefficients[[1]] + res$coefficients[[2]]*dat$x1 + res$coefficients[[3]]*1 + res$coefficients[[4]]*dat$x1*1

Vẽ các đường với cơ sở plot():

plot(dat$x1, dat$y0, type="l", xlab="x1", ylab="Estimates")
lines(dat$x1, dat$y1m, col="red")
lines(dat$x1, dat$y1p, col="blue")

Để sử dụng ggplot, bạn có thể gọi geom_line :

ggplot(dat, aes(x1, y0)) + geom_line() +
    geom_line(aes(x1, y1m), color="red") +
    geom_line(aes(x1, y1p), color="blue") +
    theme_bw() + opts(title="") + xlab("x1") + ylab("Estimates")