如何为Python计算包动态设置浮点运算精度？

我正在开发一个Python包，依赖numpy、numexp、pyfftw等库执行大量数组运算，包含多个工具脚本和一个解析.ini文件运行特定模拟的脚本。我希望通过.ini文件选择精度，让单次模拟的所有计算统一遵循该精度。

我尝试了以下逻辑，但代码无法正常运行：

config.py #

DTYPE = 'float64'

utility.py #

import numpy as np
from config import DTYPE

def create_array():
    print(DTYPE)
    arr = np.ones(10, dtype=DTYPE)
    return arr

simulation.py #

from config import DTYPE
from utility import create_array

if __name__ == '__main__':
    print(DTYPE)
    precision = get_precision(ini_file)  # 从ini文件读取精度的自定义函数
    DTYPE = precision
    create_array()

当精度设为float32时，我期望输出：

'float64'
'float32'

我清楚当前代码失效的原因：DTYPE是导入时加载的常量，运行时修改无法同步到其他模块。之前看过相关的临时解决方案，但由于用到多个库，猴子补丁（比如修改np.array默认参数）不是理想方案，而且我需要在不同模拟间灵活切换精度。

方法1：用可变字典存储配置 #

把静态常量改成可变字典，这样所有模块引用的是同一个字典实例，运行时修改能全局生效：

修改后config.py

# 用可变字典存储配置，支持运行时动态修改
config = {
    'DTYPE': 'float64'
}

修改后utility.py

import numpy as np
from config import config

def create_array():
    print(config['DTYPE'])
    arr = np.ones(10, dtype=config['DTYPE'])
    return arr

修改后simulation.py

from config import config
from utility import create_array

def get_precision(ini_file):
    # 示例：模拟从ini文件读取精度的逻辑
    return 'float32'

if __name__ == '__main__':
    print(config['DTYPE'])
    precision = get_precision('config.ini')
    config['DTYPE'] = precision
    create_array()

运行这段代码就能得到你期望的输出，因为字典是可变对象，所有模块的引用都会同步更新。

方法2：用类封装配置 #

如果需要更结构化的配置管理，可以用类属性来存储精度：

修改后config.py

class SimConfig:
    DTYPE = 'float64'

修改后utility.py

import numpy as np
from config import SimConfig

def create_array():
    print(SimConfig.DTYPE)
    arr = np.ones(10, dtype=SimConfig.DTYPE)
    return arr

修改后simulation.py

from config import SimConfig
from utility import create_array

def get_precision(ini_file):
    return 'float32'

if __name__ == '__main__':
    print(SimConfig.DTYPE)
    precision = get_precision('config.ini')
    SimConfig.DTYPE = precision
    create_array()

类属性支持运行时修改，所有模块引用的SimConfig是同一个类实例，修改后全局生效。

关键注意点 #

• 所有需要指定精度的位置，必须直接引用config['DTYPE']或SimConfig.DTYPE，不能提前赋值给局部变量（比如dtype = config['DTYPE']后再使用），否则局部变量不会同步更新。
• 对于numexp、pyfftw这类第三方库，创建数组或执行运算时要显式传入精度参数，比如numexp.evaluate('a + b', dtype=config['DTYPE'])，确保全流程精度一致。

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