在R Plotly 3D散点图中添加平行于底面的平面问题

我明白你在尝试给3D散点图添加平行于底面、z值恒为0的平面时，因为add_surface的矩阵格式要求而遇到了报错——这个函数对输入的结构确实有特定要求，我来帮你理清并解决这个问题。

问题根源：add_surface的输入格式要求 #

add_surface需要的不是你之前构造的那种3列的点矩阵，而是二维矩阵：

• 矩阵的每一行对应x轴的一个固定取值
• 矩阵的每一列对应y轴的一个固定取值
• 矩阵中[i,j]位置的元素，就是当x取第i个值、y取第j个值时的z值

你之前构造的matrix是一个11行3列的点集合，不符合这个网格结构要求，所以触发了报错。

解决方案：构造正确的网格矩阵 #

我们需要先生成x和y的取值网格，再创建对应z值全为0的二维矩阵，具体步骤如下：

1. 生成x/y的取值范围 #

先定义x和y轴的取值区间（可以根据你的散点数据范围调整，这里用你之前的-5到5）：

# 生成x和y的取值序列，length.out控制平面的精度
x_vals <- seq(-5, 5, length.out = 20)
y_vals <- seq(-5, 5, length.out = 20)

2. 创建z值全为0的二维矩阵 #

因为我们要的是z=0的平面，所以所有位置的z值都是0，矩阵维度和x/y的序列长度一致：

z_matrix <- matrix(0, nrow = length(x_vals), ncol = length(y_vals))

3. 整合到Plotly绘图代码中 #

把add_surface正确加入到你的绘图管道里，同时可以设置透明度避免遮挡散点：

n.cases <- 240
n.vars <- 4
set.seed(26)
eps <- rnorm(n.vars, 0, 1/4)
x <- matrix(rnorm(n.cases * (n.vars+2)), nrow=n.cases)
beta <- rbind(c(1,rep(0, n.vars)), c(0,rep(1, n.vars)), cbind(rep(0,n.vars), diag(eps)))
y <- x%*%beta
data <- data.frame(y)
data.1.2 <- data[,1:2]
data.1.2$X3 <- integer(nrow(data.1.2))
data.1.2$group = 'First Iteration'
data.1.3 <- data[,1:3]
data.1.3$group = 'Second Iteration'
newdf <- rbind(data.1.2, data.1.3)

library(plotly)

# 生成平面所需的网格和矩阵
x_vals <- seq(-5, 5, length.out = 20)
y_vals <- seq(-5, 5, length.out = 20)
z_matrix <- matrix(0, nrow = length(x_vals), ncol = length(y_vals))

plot_ly(x=newdf$X1, y=newdf$X2, z=newdf$X3, color = newdf$group, colors = c('#BF382A', '#0C4B8E')) %>% 
  add_markers() %>% 
  # 添加z=0的平面
  add_surface(x = x_vals, y = y_vals, z = z_matrix,
              opacity = 0.5,  # 设置透明度，避免遮挡散点
              colorscale = list(c(0,1), c('rgba(200,200,200,0.5)', 'rgba(200,200,200,0.5)')),
              showscale = FALSE) %>%  # 隐藏平面的颜色刻度条
  layout(scene = list(xaxis = list(title = 'First Dimension'), 
                      yaxis = list(title = 'Second Dimension'), 
                      zaxis = list(title = 'Third Dimension'), 
                      bgcolor = "rgb(244, 244, 248)"))

验证效果 #

运行这段代码后，你会看到一个半透明的灰色平面（z=0），它正好经过所有红色的First Iteration散点（因为这些点的z值本来就是0），同时不会遮挡蓝色的散点。

内容的提问来源于stack exchange，提问作者psteinroe