如何从meshgrid生成的球面坐标提取x,y,z并绘制带标量函数的3D曲面？

方法一：直接在原球面网格上映射标量（更简单） #

不需要额外插值，Plotly的Surface对象支持通过color参数直接绑定每个点的标量值，步骤如下：

1. 生成球坐标网格并计算x,y,z
2. 定义标量函数f(x,y,z)，直接在二维网格上计算得到对应的标量矩阵
3. 将标量矩阵传给Surface的color参数，同时可以设置colorbar来展示标量含义

示例代码：

import numpy as np 
import plotly.graph_objects as go    

# 生成球坐标网格
phi, theta = np.mgrid[0.0:np.pi:100j, 0.0:2.0 * np.pi:100j]
x = np.sin(phi) * np.cos(theta)
y = np.sin(phi) * np.sin(theta)
z = np.cos(phi)

# 定义标量函数，示例：计算每个点的x+y+z值
def f(x, y, z):
    return x + y + z

# 计算每个点的标量值，保持和x/y/z相同的二维形状
scalar_values = f(x, y, z)

# 绘制球面，用scalar_values作为颜色映射
data = [
    go.Surface(
        x=x, y=y, z=z, 
        colorscale='Electric', 
        opacity=0.8,
        color=scalar_values,  # 绑定标量值到颜色
        colorbar=dict(title='标量值f(x,y,z)')  # 显示颜色条标题
    )
]
fig = go.Figure(data=data)
fig.update_layout(title='带标量映射的3D球面')
fig.show()

方法二：提取坐标适配griddata（插值场景专用） #

如果确实需要使用scipy.interpolate.griddata，只需将二维的x/y/z数组展平为一维数组即可，步骤如下：

1. 将原二维网格的x,y,z用flatten()方法转为一维数组，形状从(100,100)变为(10000,)
2. 计算对应一维数组的标量值
3. 构建插值目标网格，调用griddata完成插值

示例代码：

import numpy as np 
import plotly.graph_objects as go    
from scipy.interpolate import griddata

# 生成球坐标网格
phi, theta = np.mgrid[0.0:np.pi:100j, 0.0:2.0 * np.pi:100j]
x = np.sin(phi) * np.cos(theta)
y = np.sin(phi) * np.sin(theta)
z = np.cos(phi)

# 展平为一维数组，适配griddata的输入格式
x_flat = x.flatten()
y_flat = y.flatten()
z_flat = z.flatten()

# 定义标量函数并计算一维的标量值
def f(x, y, z):
    return x**2 + y**2 - z**2

scalar_flat = f(x_flat, y_flat, z_flat)

# 构建插值目标网格（以x-y平面为例）
xi = np.linspace(x_flat.min(), x_flat.max(), 100)
yi = np.linspace(y_flat.min(), y_flat.max(), 100)
X, Y = np.meshgrid(xi, yi)

# 插值得到标量值的网格
scalar_interpolated = griddata((x_flat, y_flat), scalar_flat, (X, Y), method='cubic')

# 绘制插值后的曲面
fig = go.Figure(go.Surface(x=xi, y=yi, z=scalar_interpolated, colorscale='Electric'))
fig.update_layout(title='插值后的标量曲面')
fig.show()

关键说明 #

• 原球坐标生成的x/y/z是二维网格数组（形状为(100,100)），flatten()或ravel()方法可以将其转为一维数组（形状为(10000,)），完全符合griddata对输入坐标的要求
• 方法一更适合保留球面形状的标量可视化，方法二适用于将球面的标量值插值到平面网格的场景

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