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

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 .所以如果你不介意依赖私有(private)成员，你可以在那里进行查找:

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它自己想要如何缓存它的查找，它在实践中不应该是一个问题，因为它只是因为您正在加载自己的定义而失败，并且一旦您加载自己的定义，那么您已经拥有了您的信息'正在寻找定义本身!
但是，在这种情况下似乎确实存在一个错误，因为公开方法的返回值最终取决于注册表的内部状态:

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()