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

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 本身如何缓存它的查找，并且在实践中它不应该是一个问题，因为它只会失败，因为你正在加载你自己的定义，并且一次您正在加载自己的定义，好吧，您已经在定义本身中找到了要查找的信息！

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

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

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

Get base unit of a custom dimension

python

pint