Matplotlib绘制三原子分子能量曲面：如何解决不连续多边形面问题？

你的判断完全正确：原始数据点密度不足（仅484个点）是导致plot_surface出现分段、伪影的核心原因。Matplotlib的plot_surface默认只会用输入的原始点做三角剖分绘制，不会自动做平滑插值；而Excel的曲面工具默认会对稀疏数据进行插值处理，所以能得到平滑结果。要实现类似效果，必须手动对原始数据做二维插值/拟合。

具体解决方案步骤 #

以下是基于Scipy插值工具的实现思路：

1. 数据预处理
   确保你的B1、B2数据可以转换成二维网格：如果原始数据是规则网格（即B1和B2分别是等差序列的组合），直接用np.meshgrid即可；如果是不规则散点，先提取所有唯一的B1、B2值生成网格。

2. 用Scipy做二维插值
   推荐两种常用方法：

   • 径向基函数插值（Rbf）：适合不规则或规则稀疏点，插值效果平滑自然
   • 网格数据插值（griddata）：适合将散点数据插值到指定网格上

3. 代码示例

   import numpy as np
   import matplotlib.pyplot as plt
   from scipy.interpolate import Rbf, griddata
   
   # 假设你的数据是加载好的数组：b1, b2, mp3_energy（形状都是(484,)）
   b1 = np.array(...)  # 键长B1数据
   b2 = np.array(...)  # 键长B2数据
   mp3 = np.array(...) # MP3能量数据
   
   # 生成更密集的网格（提升分辨率，比如100x100）
   b1_new = np.linspace(b1.min(), b1.max(), 100)
   b2_new = np.linspace(b2.min(), b2.max(), 100)
   B1_new, B2_new = np.meshgrid(b1_new, b2_new)
   
   # 方法1：径向基函数插值
   rbf_interp = Rbf(b1, b2, mp3, function='multiquadric')
   mp3_smooth = rbf_interp(B1_new, B2_new)
   
   # 方法2：griddata插值（适合散点转规则网格）
   # mp3_smooth = griddata((b1, b2), mp3, (B1_new, B2_new), method='cubic')
   
   # 绘制平滑曲面
   fig = plt.figure()
   ax = fig.add_subplot(111, projection='3d')
   surf = ax.plot_surface(B1_new, B2_new, mp3_smooth, cmap='viridis', antialiased=True)
   ax.set_xlabel('B1 Bond Length')
   ax.set_ylabel('B2 Bond Length')
   ax.set_zlabel('MP3 Energy')
   fig.colorbar(surf)
   plt.show()
   

关键说明 #

• antialiased=True可以进一步优化曲面的视觉平滑度
• 插值方法的参数可以调整：比如Rbf的function参数可选'gaussian'、'linear'等，griddata的method可选'linear'、'cubic'，根据你的数据特性选择最贴合的
• 如果原始数据是规则网格（比如B1有22个取值，B2有22个取值，484=22×22），可以用scipy.interpolate.interp2d，效率更高

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