1

我正在使用 Python 库glom从包含列表的字典中提取数据。

这个问题是关于不了解glom如何与用户定义的函数一起工作以及如何将数据从列表移动到glom中的字典。

例如，我正在尝试将数据直接转换为字典。

# in the real world, this data has way many more fields
data = {'vessels': [{'aisStatic': {'mmsi': 1, 'name': 'name1'}},
               {'aisStatic': {'mmsi': 2, 'name': 'name2'}}],
       'timestamp': None}

# this is the glob spec to extract the data 
spec = ('vessels',[({'mmsi':'aisStatic.mmsi','name':'aisStatic.name'})])

# an example to generate a new key based on data & position
def keyfunc(i, v):
    return f"v{i}-{v['mmsi']}"

result = glom(data, spec)

wanted_result = {keyfunc(i,v): v for i, v in enumerate(result)}

想要的结果看起来像这样

{'v0-1': {'mmsi': 1, 'name': 'name1'}, 'v1-2': {'mmsi': 2, 'name': 'name2'}}

例如，我使用glom之外的理解来获得想要的结果，因为 glom 返回一个列表。

我正在寻找一种通过 glom 规范直接获得相同结果的方法。

4

1 回答 1

1

一个可行的解决方案是在enumerate函数上使用 Invoke。

from glom import merge, Invoke

data = {
    "vessels": [
        {"aisStatic": {"mmsi": 1, "name": "name1"}},
        {"aisStatic": {"mmsi": 2, "name": "name2"}},
    ],
    "timestamp": None,
}

spec = (
    "vessels",
    [{"mmsi": "aisStatic.mmsi", "name": "aisStatic.name"}],
    Invoke(enumerate).specs(T),
    [lambda x: {f"v{x[0]}-{x[1].get('mmsi')}": x[1]}],
    merge,
)

print(glom(data, spec))

返回请求的结果

{'v0-1': {'mmsi': 1, 'name': 'name1'}, 'v1-2': {'mmsi': 2, 'name': 'name2'}}
于 2020-02-26T14:37:06.667 回答