如何修改已创建的pint.Quantity对象的_magnitude字段？

问题描述 #

通过性能分析发现，代码瓶颈在于频繁调用pint.Quantity构造函数创建新对象并赋值给固定长度列表：

import pint
ureg = pint.UnitRegistry()
...
quantity = pint.Quantity(-2.5, ureg.kcal / ureg.mol)
quantities[i] = quantity

由于列表中所有Quantity对象的单位固定为kcal/mol，尝试原地修改数值时，直接设置magnitude属性触发AttributeError，而修改私有字段_magnitude虽能提升性能，但担心引发潜在问题。

可行解决方案 #

1. 基于单位固定场景的私有字段修改（性能最优，需注意版本兼容） #

在单位完全固定的前提下，直接修改_magnitude是安全的——因为Pint中与单位相关的衍生字段（如转换系数、单位标识）不会因数值变化而改变，仅数值存储字段_magnitude需要更新。代码示例：

# 初始化列表：提前创建好所有单位为kcal/mol的Quantity对象
quantities = [pint.Quantity(0, ureg.kcal/ureg.mol) for _ in range(fixed_length)]

# 原地修改数值
quantities[i]._magnitude = -2.5

注意：此方式依赖Pint的内部实现，升级Pint版本时需验证该字段是否存在或逻辑是否变化。

2. 分离数值存储与Quantity包装（官方兼容，性能接近最优） #

将数值单独存储在数组（如numpy数组）中，仅在需要使用Quantity对象时进行包装，完全避免修改Quantity实例：

import numpy as np
ureg = pint.UnitRegistry()

# 初始化数值数组
magnitudes = np.zeros(fixed_length)
target_unit = ureg.kcal / ureg.mol

# 修改数值直接操作数组
magnitudes[i] = -2.5

# 需要Quantity时再包装
current_quantity = pint.Quantity(magnitudes[i], target_unit)

这种方式符合官方API规范，无内部实现依赖，同时性能损失极小。

3. 官方兼容的间接修改方式（安全性最高，性能略逊） #

如果完全不想依赖内部字段，可以通过创建临时Quantity对象并复制状态的方式，但性能会比直接修改私有字段差：

temp_q = pint.Quantity(-2.5, ureg.kcal/ureg.mol)
quantities[i].__dict__.update(temp_q.__dict__)

但此方式本质还是复制状态，性能提升有限，仅适合对内部实现敏感的场景。

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