Add a regression plane to a 3d scatter plot in a flat

Question

I'm looking to take advantage of the excellent features in Plotly, but it's hard for me to figure out how to add a regression plane to a 3D graphic. Here is an example of how to get started with a 3D plot, does anyone know how to take the next step and add a plane?

library(plotly) data(iris) iris_plot <- plot_ly(my_df, x = Sepal.Length, y = Sepal.Width, z = Petal.Length, type = "scatter3d", mode = "markers") petal_lm <- lm(Petal.Length ~ 0 + Sepal.Length + Sepal.Width, data = iris) 

Answer 1

You need to try points based on the forecast object created from your lm call. This creates an object-like surface.

At this moment, we have petal_lm_surface , which has a z value for all x and y that we want to display. Now we just need to create a base graph (points) by adding color and text for each view:

 hcolors=c("red","blue","green")[my_df$Species] iris_plot <- plot_ly(my_df, x = Sepal.Length, y = Sepal.Width, z = Petal.Length, text = Species, type = "scatter3d", mode = "markers", marker = list(color = hcolors)) 

and then add the surface:

 iris_plot <- add_trace(last_plot = iris_plot, z = petal_lm_surface, x = axis_x, y = axis_y, type = "surface") iris_plot