R 从 predict() 获取数字矩阵

Question

我有以下代码:

fit_lm=lm(z~x+y)

mix <- 2
max <- 12
miy <- 2
may <- 12

griddf <- expand.grid(x = seq(mix,max, length.out = 10),
                      y = seq( miy,may,length.out = 10))


Prediction_data <- data.frame(griddf)
colnames(Prediction_data) <- c("x", "y")
coordinates(Prediction_data ) <- ~ x + y

terrain_lm <- predict(fit_lm, Prediction_data)

我希望 terrain_lm 是一个我可以使用的数字矩阵

fig <- plot_ly()

fig <- fig %>%  add_surface(terrain_lm)

但我得到一个包含 100 个元素的一维数组。

Answer 1

predict 的结果是一个向量。您需要将它添加到 x 和 y 值，然后使用 xtabs 转换为适合曲面图的矩阵。

library(plotly)

#test data
x <- runif(20, 4, 10)
y <- runif(20, 3, 6)
z <- 3*x+y +runif(20, 0, 2)

fit_lm <- lm(z~x+y)


mix <- 2
max <- 12
miy <- 2
may <- 12

griddf <- expand.grid(x = seq(mix,max, length.out = 10),
                      y = seq( miy,may,length.out = 10))


terrain_lm <- data.frame(griddf)
terrain_lm$z <- predict(fit_lm, terrain_lm)

fig <- plot_ly(z = ~xtabs(z ~ x + y, data = terrain_lm))
fig <- fig %>%  add_surface()