如何在R的Plotly中添加绕X轴旋转的回转体曲面（按X着色）

我来帮你把这个回转体的实现补全，并且实现按X轴方向着色的效果，直接看完整可运行的代码和细节解释：

library(dplyr)
library(plotly)

# 定义随x变化的半径函数
r <- function(x) x^2
# 设置旋转的x轴区间
int <- c(1, 3)
# x轴方向的采样点数
nx <- 20
# 旋转圆周方向的采样点数（越多曲面越平滑）
ntheta <- 36

# 生成x采样点并计算对应半径
coords <- tibble(x = seq(int[1], int[2], length.out = nx)) %>% 
  mutate(r = r(x))

# 生成旋转角度序列（转成弧度）
theta <- seq(0, 2*pi, length.out = ntheta)

# 构造Plotly surface需要的二维矩阵格式数据
# x矩阵：每一行重复相同的x值，对应不同旋转角度
x_matrix <- matrix(rep(coords$x, ntheta), nrow = nx, ncol = ntheta)
# y坐标：r*cos(theta)
y_matrix <- matrix(coords$r %o% cos(theta), nrow = nx, ncol = ntheta)
# z坐标：r*sin(theta)
z_matrix <- matrix(coords$r %o% sin(theta), nrow = nx, ncol = ntheta)

# 初始化绘图并添加回转体曲面，按X轴着色
p <- plot_ly() %>%
  add_surface(
    x = x_matrix,
    y = y_matrix,
    z = z_matrix,
    # 绑定到coloraxis，实现按x值着色
    coloraxis = "coloraxis"
  ) %>%
  # 配置coloraxis：颜色映射基于x的取值范围，搭配颜色条说明
  layout(
    coloraxis = list(
      colorscale = "Viridis",
      cmin = min(coords$x),
      cmax = max(coords$x),
      colorbar = list(title = "X轴值")
    ),
    scene = list(
      xaxis = list(title = "X"),
      yaxis = list(title = "Y"),
      zaxis = list(title = "Z")
    )
  )

# 展示绘图
p

关键细节说明： #

• 数据格式适配：Plotly的add_surface要求输入的x、y、z都是二维矩阵，所以我们把每个x值重复对应所有旋转角度生成x_matrix；y和z则通过外积（%o%）快速计算每个x与theta组合的坐标。
• 按X轴着色核心：通过coloraxis参数把曲面颜色绑定到x的取值，再在layout里配置色阶范围和颜色条，让曲面颜色随x轴数值渐变。
• 采样点数调整：nx和ntheta的值越大，回转体表面越平滑，但计算速度会稍慢，你可以根据需求灵活调整。

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