Группировка строк из набора в словарь питона

Question

Вход:

>>> foo = {("A", "B"), ("A", "G"), ("A", "H"), 
...   ("B", "C"), ("B", "H"), 
...   ("C", "D"), 
...   ("D", "E"), ("D", "G"), ("D", "H"), 
...   ("E", "F"), ("E", "G"), ("E", "H"), 
...   ("F", "I"), ("F", "J"), ("F", "K"), 
...   ("G", "H")} 

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

Фактический выход:

>>> {k:set(x[1] for x in v) for k,v in itertools.groupby(sorted(foo), key=lambda x: x[0])} 
{'B': {'C', 'H'}, 'E': {'H', 'G', 'F'}, 'D': {'E', 'G', 'H'}, 'A': {'B', 'G', 'H'}, 'C': {'D'}, 'G': {'H'}, 'F': {'I', 'K', 'J'}} 

Однако, это мой желаемый результат:

foo1 = {"A": {"B", "G", "H"}, 
     "B": {"A", "H", "C"}, 
     "C": {"B", "D"}, 
     "D": {"C", "E", "G", "H"}, 
     "E": {"D", "F", "G", "H"}, 
     "F": {"E", "I", "J", "K"}, 
     "G": {"A", "D", "E", "H"}, 
     "H": {"A", "B", "D", "E", "G"}, 
     "I": {"F"}, 
     "J": {"F"}, 
     "K": {"F"} 
     } 

Я думал, с перестановками или комбинациями? Но я до сих пор не знаю, как это сделать. Спасибо заранее. Редактировать: Добавлено изображение того, как связаны строки.

Answer 1

OP просил, чтобы это было сделано с пониманием, так вот безвозмездное один лайнер:

(Пожалуйста, пожалуйста, пожалуйста, не делайте это так)

python3 
Python 3.5.2+ (default, Sep 22 2016, 12:18:14) 
[GCC 6.2.0 20160927] on linux 
Type "help", "copyright", "credits" or "license" for more information. 
>>> import itertools 
>>> import pprint 
>>> 
>>> foo = {("A", "B"), ("A", "G"), ("A", "H"), 
...   ("B", "C"), ("B", "H"), 
...   ("C", "D"), 
...   ("D", "E"), ("D", "G"), ("D", "H"), 
...   ("E", "F"), ("E", "G"), ("E", "H"), 
...   ("F", "I"), ("F", "J"), ("F", "K"), 
...   ("G", "H")} 
>>> 
>>> foo1 = {k: {x[1] for x in v} for k, v in itertools.groupby(sorted(foo | {(a, b) for b, a in foo}), key=lambda x: x[0])} 
>>> pprint.pprint(foo1) 
{'A': {'H', 'B', 'G'}, 
'B': {'C', 'H', 'A'}, 
'C': {'D', 'B'}, 
'D': {'C', 'E', 'H', 'G'}, 
'E': {'D', 'H', 'F', 'G'}, 
'F': {'K', 'J', 'E', 'I'}, 
'G': {'D', 'E', 'H', 'A'}, 
'H': {'D', 'E', 'B', 'G', 'A'}, 
'I': {'F'}, 
'J': {'F'}, 
'K': {'F'}} 

Answer 2

Вы можете использовать обычный словарь, но defaultdict делает это немного проще.

from collections import defaultdict 

d=defaultdict(set) 
for key, value in foo: 
    f[key].add(value) 

При регулярном Dict:

d = {} 
for key, value in foo: 
    if key not in d: 
     d[key]=set() 
    d[key].add(value) 

Answer 3

Что-то вроде этого?

foo = {("A", "B"), ("A", "G"), ("A", "H"), ("B", "C"), ("B", "H"), ("C", "D"), ("D", "E"), ("D", "G"), ("D", "H"), ("E", "F"), ("E", "G"), ("E", "H"), ("F", "I"), ("F", "J"), ("F", "K"), ("G", "H")} 

from collections import defaultdict 

result = defaultdict(set) 

for key1, key2 in foo: 
    result[key1].add(key2) 
    result[key2].add(key1) 

from pprint import pprint 
pprint(result) 

Который, в отличие от других ответов до сих пор, дает желаемый результат, хотя и неупорядоченное:

{'A': {'B', 'G', 'H'}, 
'B': {'A', 'C', 'H'}, 
'C': {'B', 'D'}, 
'D': {'G', 'C', 'H', 'E'}, 
'E': {'G', 'D', 'H', 'F'}, 
'F': {'I', 'J', 'K', 'E'}, 
'G': {'A', 'D', 'H', 'E'}, 
'H': {'B', 'A', 'G', 'D', 'E'}, 
'I': {'F'}, 
'J': {'F'}, 
'K': {'F'}}