Início / coding / Perguntas / 77052659
Accepted
Fravadona
Asked: 2023-09-06 22:02:20 +0800 CST

compreensão de lista python: lista de ditados para ditados de listas com interseção de chave

  • 772

Tenho uma lista com um número variável de dicionários, por exemplo:

var = [ {'a': 1, 'b': 2}, {'b': 20, 'a': 10, 'c': 30}, {'c': 300, 'a': 100} ]

Preciso extrair as chaves que são comuns a todos os dictos, fazer uma lista de seus valores associados, criar um novo dict a partir dele e armazená-lo na mesma variável:

O resultado esperado seria:

var = { 'a': [1, 10, 100] }

Posso encontrar a interseção das chaves com:

[k for k in var[0] if all(k in d for d in var[1:])]

Mas como você pode fazer o resto da transformação?

python

2 respostas

  1. Depois de ter as chaves comuns, você pode usar uma compreensão de ditado e definir o valor como uma compreensão de lista aninhada:

    {k: [v[k] for v in var] for k in set.intersection(*map(set, var))}

    Observação:

    Para encontrar as chaves comuns você pode transformar os dictos em conjuntos (isso fará um conjunto usando apenas as chaves) e então usar a interseção de conjuntos:

    • 4
  2. Best Answer

    Depois de conhecer as chaves de seu interesse, apenas itere e puxe-as conforme você avança:

    from collections import defaultdict

new_var = defaultdict(list)
for d in var:
    for k in common_keys:
        new_var[k].append(d[k])

new_var = dict(new_var)  # Optionally convert back to plain dict to avoid autovivification

    Uma linha única também é possível (já que você garantiu que as chaves existem em todos dictos s), é um pouco feio, considerando a quantidade de significado que ela coloca em uma única linha:

    new_var = {k: [d[k] for d in var] for k in common_keys}

    Nesse caso, o one-liner está bom, mas é um pouco menos flexível se você precisar modificá-lo; o loop explícito é mais fácil de ajustar, porém mais detalhado.

    Nota lateral: existe uma maneira mais simples/rápida de calcular as chaves comuns:

    common_keys = set(var[0]).intersection(*var[1:])

    Isso converte as chaves do primeiro dictem a sete permite que seto intersectionmétodo produza a interseção em uma única chamada na camada C (não é significativamente diferente na forma como opera em relação ao seu código, mas evita uma tonelada de sobrecarga do intérprete) .

    Você pode até fazer com que ele lide silenciosamente com um vazio var, alterando-o para:

    common_keys = set(*var[:1]).intersection(*var[1:])

    que produzirá um vazio setfor common_keysif varcontains no dicts (a escolha depende do cenário; se esvaziar varfor um erro, morrer ruidosamente é melhor do que fazer um trabalho inválido silenciosamente).

    Combinar as duas partes permitiria que você alcançasse a solução verdadeiramente horrível como um all-in-one-liner completo:

    {k: [d[k] for d in var] for k in set(var[0]).intersection(*var[1:])}

# Or to silently accept empty var:

{k: [d[k] for d in var] for k in set(*var[:1]).intersection(*var[1:])}

    mas esse é um código realmente horrível, então sugiro dividi-lo um pouco.

    • 3

relate perguntas