2

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

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})>
4

1 回答 1

2

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

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

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

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()
于 2021-01-06T09:18:19.597 回答