Matplotlib mplot3d：填充边缘以获取‘凸面’的技术求助

嗨，我来帮你实现这个凸面填充的需求！你现在的曲面图已经能正确展示现有网格数据，但要补上边缘的四个三角形让它变成凸面，我们可以通过手动添加3D多边形的方式来实现，同时完美匹配原曲面的配色。

核心思路 #

原有的plot_surface只绘制了有规则网格数据的部分，我们需要手动定义四个边缘三角形的顶点，然后用Poly3DCollection把这些三角形添加到3D轴上，并且复用原曲面的颜色映射和归一化规则，保证颜色统一。

修改后的完整代码 #

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.art3d import Poly3DCollection

def getMeshGrid(dat):
    x = dat[:,0]; y = dat[:,1]; z = dat[:,2]
    xVal, xIdx = np.unique(x, return_inverse=True)
    yVal, yIdx = np.unique(y, return_inverse=True)
    zVal = np.empty(xVal.shape+yVal.shape)
    zVal.fill(np.nan)
    zVal[xIdx, yIdx] = z
    xx, yy = np.meshgrid(np.unique(x), np.unique(y))
    zz = zVal.T
    return (xx, yy, zz)

dat = np.loadtxt("tmp.dat")
X, Y, Z = getMeshGrid(dat)

fig = plt.figure()
axe = fig.add_subplot(111, projection='3d')

# 绘制原曲面，并保存颜色映射和归一化参数
surf = axe.plot_surface(X, Y, Z, cmap='viridis')
norm = surf.norm
cmap = surf.cmap

# 定义需要填充的4个边缘三角形（对应你描述的凸面边缘）
triangles = [
    # 底部y=1平面的三角形：(1,1,1)、(3,1,1)、(2,2,5)
    [(1,1,1), (3,1,1), (2,2,5)],
    # 顶部y=3平面的三角形：(1,3,1)、(3,3,1)、(2,2,5)
    [(1,3,1), (3,3,1), (2,2,5)],
    # 左侧x=1平面的三角形：(1,1,1)、(1,3,1)、(2,2,5)
    [(1,1,1), (1,3,1), (2,2,5)],
    # 右侧x=3平面的三角形：(3,1,1)、(3,3,1)、(2,2,5)
    [(3,1,1), (3,3,1), (2,2,5)]
]

# 创建3D多边形集合
poly3d = Poly3DCollection(triangles, alpha=1.0)
# 匹配原曲面的配色：根据每个三角形的最大z值计算颜色
colors = [cmap(norm(np.max([p[2] for p in tri]))) for tri in triangles]
poly3d.set_facecolors(colors)

# 将填充的三角形添加到3D坐标轴
axe.add_collection3d(poly3d)

# 可选：设置坐标轴标签让图表更清晰
axe.set_xlabel('X')
axe.set_ylabel('Y')
axe.set_zlabel('Z')

plt.show()

关键细节说明 #

1. 颜色匹配：通过surf.norm和surf.cmap获取原曲面的颜色归一化规则和配色方案，这样填充的三角形颜色会和原曲面完全统一，不会出现色差。
2. 三角形定义：我根据你描述的边缘点，定义了四个连接边缘和顶部中心点(2,2,5)的三角形，刚好组成你需要的凸面结构。如果你的需求有微调，直接修改triangles列表里的顶点坐标即可。
3. 灵活性：这种方式不需要修改原始数据或网格生成逻辑，直接在现有绘图基础上补充填充面，操作简单且可控。

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