озадаченный получение значения ключа из понимания - PullRequest
Новая
       14

озадаченный получение значения ключа из понимания

1 голос
/ 22 марта 2019

Я пытаюсь выяснить, как получить значения из понимания, которое почти работает.Из этих данных .. 

{'rock': {}, 'coal1': {'gold1': {'data': ['g1']}}, 'coal2': {'gold3': {'data': ['g3']}, 'gold2': {'data': ['g2']}}}

.. я пытаюсь извлечь данные золотого ключа в простой дикт.то есть диктат из { gold : {'data' : [...]} } предметов, по сути, вырывающий угольные ключи из данных.

другими словами из этого .. 

{
  "coal2": {
    "gold3": {
      "data": [
        "g3"
      ]
    },
    "gold2": {
      "data": [
        "g2"
      ]
    }
  },
  "coal1": {
    "gold1": {
      "data": [
        "g1"
      ]
    }
  },
  "rock": {}
}

в этот формат 

{
  "gold3": {
    "data": [
      "g3"
    ]
  },
  "gold1": {
    "data": [
      "g1"
    ]
  },
  "gold2": {
    "data": [
      "g2"
    ]
  }
}

Это почти работает.Это избавляет от камня. 

>>> {k:d for k,d in data.items() if k != 'rock'}
{'coal2': {'gold3': {'data': ['g3']}, 'gold2': {'data': ['g2']}}, 'coal1': {'gold1': {'data': ['g1']}}}

А получение значений избавляет от угольных ключей. 

>>> [v for v in {k:d for k,d in data.items() if k != 'rock'}.values()]
[{'gold3': {'data': ['g3']}, 'gold2': {'data': ['g2']}}, {'gold1': {'data': ['g1']}}]

Но я не могу понять, как получить из этого 

>>> for i in [v for v in {k:d for k,d in data.items() if k != 'rock'}.values()] : print(i)
...
{'gold3': {'data': ['g3']}, 'gold2': {'data': ['g2']}}
{'gold1': {'data': ['g1']}}

до нужной структуры.И было бы мило, если бы все это можно было сделать с пониманием.Кто-нибудь знает, как это сделать?

РЕДАКТИРОВАТЬ: Оба ответа было удивительно, и я хотел бы принять оба.Мне нравится ничего не импортировать, но я принимаю версию @blhsing itertools только потому, что ее легче понять, и она намного лучше.Кстати, камень должен быть отброшен, даже если он имеет значения, поэтому я не мог обойти if k != 'rock'.Итак, вот результаты и ... Спасибо, ребята.

>>> import timeit
>>> data = {'rock': {'type':'pebble'}, 'coal1': {'gold1': {'data': ['g1']}}, 'coal2': {'gold3': {'data': ['g3']}, 'gold2': {'data': ['g2']}}}
>>> timeit.timeit( "dict(kv for x in (v for v in {k:d for k,d in data.items() if k != 'rock'}.values()) for kv in x.items())" ,  setup="from __main__ import data")            
2.6714617270044982
>>>
>>> timeit.timeit( "dict(chain.from_iterable(g.items() for g in {k:d for k,d in data.items() if k != 'rock'}.values()))" , setup="from __main__ import data; from itertools import chain")
2.22612579818815
>>>

Ответы [ 2 ]

1 голос
/ 22 марта 2019

Вы можете использовать выражение генератора, которое выводит элементы поддиктов значений основного диктата, и использовать itertools.chain.from_iterable, чтобы соединить элементы и передать их конструктору dict: 

from itertools import chain
dict(chain.from_iterable(g.items() for g in d.values()))

так что дано: 

d = {'rock': {}, 'coal1': {'gold1': {'data': ['g1']}}, 'coal2': {'gold3': {'data': ['g3']}, 'gold2': {'data': ['g2']}}}

это возвращает: 

{'gold3': {'data': ['g3']}, 'gold2': {'data': ['g2']}, 'gold1': {'data': ['g1']}}
1 голос
/ 22 марта 2019

Просто нужно сделать list из dict до dict (исправление кода с добавлением второй строки) 

l=[v for v in {k:d for k,d in d.items() if k != 'rock'}.values()] # here is your own code 
newd=dict(kv for x in l for kv in x.items())
newd
Out[431]: 
{'gold1': {'data': ['g1']},
 'gold2': {'data': ['g2']},
 'gold3': {'data': ['g3']}}

с одной строкой 

dict(v for d in d.values() for v in d.items()) # d is your dict
Out[436]: 
{'gold1': {'data': ['g1']},
 'gold2': {'data': ['g2']},
 'gold3': {'data': ['g3']}}
...