Đây là một autoplot
phương thức cho lớp "cơ sở" (mà cả bs và ns đều thừa hưởng từ):
library(ggplot2)
library(magrittr)
library(reshape2)
library(stringr)
autoplot.basis <- function(basis, n=1000) {
all.knots <- sort(c(attr(basis,"Boundary.knots") ,attr(basis, "knots"))) %>%
unname
bounds <- range(all.knots)
knot.values <- predict(basis, all.knots) %>%
set_colnames(str_c("S", seq_len(ncol(.))))
newx <- seq(bounds[1], bounds[2], length.out = n+1)
interp.values <- predict(basis, newx) %>%
set_colnames(str_c("S", seq_len(ncol(.))))
knot.df <- data.frame(x=all.knots, knot.values) %>%
melt(id.vars="x", variable.name="Spline", value.name="y")
interp.df <- data.frame(x=newx, interp.values) %>%
melt(id.vars="x", variable.name="Spline", value.name="y")
ggplot(interp.df) +
aes(x=x, y=y, color=Spline, group=Spline) +
geom_line() +
geom_point(data=knot.df) +
scale_color_discrete(guide=FALSE)
}
Điều này cho phép bạn chỉ cần gọi autoplot
một đối tượng ns hoặc bs. Lấy ví dụ của jbowman:
library(splines)
x <- seq(0, 1, by=0.001)
spl <- bs(x,df=6)
autoplot(spl)
sản xuất:
Chỉnh sửa: Điều này sẽ được bao gồm trong phiên bản tiếp theo của gói ggfortify: https://github.com/sinhrks/ggfortify/pull/129 . Sau đó, tôi tin rằng tất cả những gì bạn cần là:
library(splines)
library(ggfortify)
x <- seq(0, 1, by=0.001)
spl <- bs(x,df=6)
autoplot(spl)
matplot
hàm thay vì lặp qua các cột.