运行立方体网格生成代码时遇pygmsh无generate_mesh属性报错

问题重现 #

我尝试用pygmsh生成立方体网格，代码如下：

import pygmsh as pg
import numpy as np
def generate():
    geom = pg.Geometry()
    geom.add_box(0, 1, 0, 1, 0, 1, 0.05)
    return geom
if __name__ == '__main__':
    import meshio
    points, cells = pg.generate_mesh(generate())
    meshio.write('cube.vtu', points, cells)

结果运行时抛出了这个错误：

Traceback (most recent call last):
  File "C:/Users/200498/PycharmProjects/untitled/pygmsh.py", line 17, in <module>
    points, cells = pg.generate_mesh(generate())
AttributeError: module 'pygmsh' has no attribute 'generate_mesh'

问题根源 #

这个坑我之前踩过！主要是因为pygmsh的API在v5.x版本做了大幅调整：

• 在v4.x及更早的版本中，generate_mesh是直接挂载在pygmsh模块下的全局函数；
• 但升级到v5.x之后，这个函数被移到了pygmsh.generate子模块里，或者你也可以直接通过Geometry实例来调用生成方法。

另外，还有一个容易忽略的点：你的脚本文件名是pygmsh.py，这会导致Python导入时优先加载你自己的脚本，而不是真正的pygmsh库，这也可能触发这个错误！

解决方案 #

方案1：适配新版本pygmsh（推荐） #

如果你的pygmsh是v5.x+，修改代码有两种方式：

方式A：导入正确的generate_mesh函数

import pygmsh as pg
import numpy as np
import meshio

def generate():
    geom = pg.Geometry()
    # 新版本建议显式指定mesh_size参数名，可读性更好
    geom.add_box(0, 1, 0, 1, 0, 1, mesh_size=0.05)
    return geom

if __name__ == '__main__':
    # 从正确的子模块导入generate_mesh
    from pygmsh.generate import generate_mesh
    points, cells = generate_mesh(generate())
    meshio.write('cube.vtu', points, cells)

方式B：直接调用Geometry实例的生成方法

更简洁的写法是直接用Geometry对象的generate_mesh方法：

import pygmsh as pg
import numpy as np
import meshio

def generate():
    geom = pg.Geometry()
    geom.add_box(0, 1, 0, 1, 0, 1, mesh_size=0.05)
    return geom

if __name__ == '__main__':
    geom = generate()
    # 直接调用实例方法
    points, cells = geom.generate_mesh()
    meshio.write('cube.vtu', points, cells)

方案2：降级到旧版本 #

如果你不想改代码，只想快速运行，可以降级到v4.x的稳定版本：

pip install pygmsh==4.11.1

这个版本保留了原来的全局generate_mesh函数，你的原始代码可以直接跑起来。

关键注意事项 #

一定要把你的脚本文件名从pygmsh.py改成别的（比如cube_mesh_generator.py），否则Python会一直导入你自己的脚本，永远找不到真正的pygmsh库！

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