获取自定义维度的基本单位

Question

假设我定义了一个新维度并沿该维度定义了新单位。 在此示例中，我使用货币和合成汇率，但也可以是任何其他自定义维度：

import pint 
ureg = pint.UnitRegistry()
Q_ = ureg.Quantity
import io
ctx_def = io.StringIO("""\
EUR = [currency]
DKK = 0.14 EUR
JPY = 0.01 EUR
USD = 0.9 EUR
GBP = 1.1 EUR
""")
ureg.load_definitions(ctx_def)

在这里， EUR是基本单位，转换到这个基本单位可以正常工作：

Q_(42, "JPY").to_base_units()
# returns 0.42 EUR as expected

我的问题是：给定这个单位注册表，并给定自定义维度名称作为输入，即"[currency]" ，我如何获得基本单位"EUR" ？

---

如果它是一个内置维度，例如[mass] ，那么我可以这样做（不优雅，但有效）：

ureg.get_base_units(list(ureg.get_compatible_units("[mass]"))[0])[1]
# returns "kilogram"

但是，此技巧不适用于我的自定义维度[currency] ：

ureg.get_base_units(list(ureg.get_compatible_units("[currency]"))[0])[1]
# raises:
# KeyError: <UnitsContainer({'[currency]': 1})>

Answer 1

一种选择是只查找所有包含单元的维度并取第一个匹配的单元：

from pint.registry import UnitsContainer

def root_unit(dim):
    uc = UnitsContainer({dim: 1})
    return next(ureg.get_root_units(unit)[1] for unit in ureg if ureg.get_dimensionality(unit) == uc)

print(root_unit('[currency]'))  # EUR
print(root_unit('[mass]'))      # gram

另请注意，单位映射到底层RegistryCache中的维度，存储为ureg._cache 。 所以如果你不介意依赖私人成员，你可以在那里进行查找：

In [14]: [k for k, v in ureg._cache.dimensionality.items() if v == UnitsContainer({'[currency]': 1})]
Out[14]:
[<ParserHelper(1, {'EUR': 1})>,
 <ParserHelper(1, {'USD': 1})>,
 <ParserHelper(1, {'DKK': 1})>,
 <ParserHelper(1, {'JPY': 1})>,
 <ParserHelper(1, {'GBP': 1})>]

特别是，将[currency]成EUR相当于

def root_unit(dim):
    uc = UnitsContainer({dim: 1})
    return next(ureg.get_root_units(k)[1] for k, v in ureg._cache.dimensionality.items() if v == uc)

ureg._build_cache()
print(root_unit('[currency]'))  # EUR
print(root_unit('[mass]'))      # gram

如果维度本身是RegistryCache.root_units中的一个键，这一切都会更简单一些，但仍然可能更糟。

但是请注意必要的ureg._build_cache() ：缓存仅在您加载定义之前构建。 先验，这很公平，因为它取决于UnitRegistry自己如何缓存其查找，并且在实践中它不应该成为问题，因为它只会失败，因为您正在加载自己的定义，并且一旦您正在加载您自己的定义，好吧，您已经在定义本身中找到了您要查找的信息！

但是，在这种情况下似乎确实存在一个错误，因为公开方法的返回值最终取决于注册表的内部 state：

In [26]: ureg.get_compatible_units('EUR')  # Fails
---------------------------------------------------------------------------
KeyError

In [27]: ureg._build_cache()

In [28]: ureg.get_compatible_units('EUR')  # No longer fails
Out[28]: frozenset()