python - 使用 glom 将字典列表的字段提取到新字典中

Question

我有以下高度简化的结构

elements = [{"id": "1", "counts": [1, 2, 3]},
            {"id": "2", "counts": [4, 5, 6]}]

我希望能够使用 glom 构建，形式为 {<id>: <counts[pos]>} 的新词典，例如对于 pos = 2 :

{"1": 3, "2": 6}

或者一个列表/元组的元组

[("1",3), ("2", 6)]

使用 dict comprehension很简单，但是数据结构比较复杂，我想动态指定要提取的内容。前面的例子是我想要实现的最简单的事情。

经过一段时间我设法解决了如下问题

from glom import glom, T

elements = [{"id": "1", "counts": [1,2,3]},{"id": "2", "counts": [4,5,6]}]

def extract(elements, pos):
    extracted = glom(elements, ({"elements": [lambda v: (v["id"], v["counts"][pos])]}, T))
    return dict(extracted["elements"])

但这需要调用 dict .跳过字典间接寻址的细微变化是

def extract(elements, pos):
    extracted = glom(elements, (([lambda v: {v["id"]: v["counts"][pos]}]), T))
    return {k: v for d in extracted for k, v in d.items()}

现在，我可以使用 merge函数调用了 glom 的返回值打电话

def extract(elements, pos):
    return merge(glom(elements, (([lambda v: {v["id"]: v["counts"][pos]}]), T)))

我对此很满意，但是有没有更好的方法来做到这一点？更好的意思是构建一个更清洁的可调用规范？最终，我希望能够在运行时以用户友好的方式定义字典的值，即 v["counts"][pos] .

对这个想法的改进是使用可调用对象来调用内部字典的值

def counts_position(element, **kwargs):
    return element["counts"][kwargs["pos"]]

def extract(elements, func, **kwargs):
    return merge(glom(elements, (([lambda v: {v["id"]: func(v, **kwargs)}]), T)))

extract(values, counts_position, pos=2)

这样，可以从外部控制从每个元素中提取的内容。

Answer 1

要将每个具有 id 的字典列表转换为以 id 为键的字典，您可以使用简单的字典理解:

{t["id"]: glom.glom(t, "counts.2") for t in elements}

或者，如果你想为此使用 glom，use glom.Merge along with glom.T :

glom.glom(elements, glom.Merge([{T['id']: 'counts.2'}])))

为了避免 lambda，您可以将 pos 参数插入到规范字符串中，例如'counts.%s' % pos.