动态即时编译Cython：导入.pyd模块后无法编译自定义函数的问题

你的核心问题是Windows系统下已导入的.pyd模块会被系统锁定，导致编译过程无法覆盖或生成新的模块文件，下面给你几个实用的解决方案，按推荐程度排序：

1. 分离编译逻辑与运行逻辑（最稳定） #

把生成Cython代码、编译.pyd的步骤单独放在一个脚本里，主脚本只负责导入编译好的模块并执行计算。这样编译时主脚本没有运行，自然不会锁定模块文件。

示例代码 #

编译脚本 compile_custom_func.py

from setuptools import setup
from Cython.Build import cythonize
import os

# 替换成用户输入的自定义函数表达式，比如"x**2"、"math.exp(x)"等
user_expr = "x**2"

# 生成Cython源码内容
pyx_content = f"""
import math

def compute_coords(x_list):
    result = []
    for x in x_list:
        result.append({user_expr})
    return result
"""

# 写入.pyx文件
with open("custom_func.pyx", "w", encoding="utf-8") as f:
    f.write(pyx_content)

# 编译为.pyd模块（Windows平台）
setup(
    ext_modules=cythonize("custom_func.pyx", language_level="3"),
)

运行脚本 main.py

import custom_func

# 测试自定义函数计算坐标
x_coords = [1.0, 2.0, 3.0, 4.0]
y_coords = custom_func.compute_coords(x_coords)
print(f"计算结果：{y_coords}")

使用步骤 #

1. 先运行编译脚本：python compile_custom_func.py build_ext --inplace
2. 再运行主脚本：python main.py

2. 同脚本内先卸载旧模块再编译（适合简单场景） #

如果必须在同一个脚本里完成编译+运行，可以先检查模块是否已导入，从sys.modules中移除并删除旧的.pyd文件，再进行编译。不过Windows下偶尔会出现文件锁定无法删除的情况，需要注意。

import sys
import os
import importlib
from setuptools import setup
from Cython.Build import cythonize

user_expr = "math.exp(x)"
module_name = "custom_func"

# 生成.pyx文件
with open(f"{module_name}.pyx", "w", encoding="utf-8") as f:
    f.write(f"""
import math

def compute_coords(x_list):
    return [{user_expr} for x in x_list]
""")

# 清理旧模块
if module_name in sys.modules:
    del sys.modules[module_name]
# 删除旧的.pyd文件
for file in os.listdir():
    if file.startswith(module_name) and file.endswith(".pyd"):
        try:
            os.remove(file)
        except PermissionError:
            print("错误：旧模块文件被占用，请关闭相关进程后重试")
            sys.exit(1)

# 编译模块
setup(
    ext_modules=cythonize(f"{module_name}.pyx", language_level="3"),
)

# 导入新编译的模块并测试
custom_func = importlib.import_module(module_name)
print(custom_func.compute_coords([0, 1, 2, 3]))

3. 使用动态模块名避免冲突（适合频繁更新函数的场景） #

每次编译时生成唯一的模块名（比如加时间戳或哈希值），这样不会覆盖旧模块，也不会因为旧模块被加载导致编译失败。

import time
import math
import importlib
from setuptools import setup
from Cython.Build import cythonize

user_expr = "x * math.sin(x)"
# 生成唯一模块名
module_name = f"custom_func_{int(time.time())}"

# 生成Cython代码
pyx_content = f"""
import math

def compute_coords(x_list):
    result = []
    for x in x_list:
        result.append({user_expr})
    return result
"""

with open(f"{module_name}.pyx", "w", encoding="utf-8") as f:
    f.write(pyx_content)

# 编译模块
setup(
    ext_modules=cythonize(f"{module_name}.pyx", language_level="3"),
)

# 导入并使用模块
custom_func = importlib.import_module(module_name)
print(custom_func.compute_coords([0, math.pi/2, math.pi]))

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