Python: Mögliche Kombinationen für Elemente in einer Liste von Listen

Question

Angenommen, ich habe eine verschachtelte Liste, die wie folgt aussieht:

nested_list = [[a,b,1],[c,d,3],[a,f,8],[a,c,5]] 

Jetzt mag ich zwischen dem zweiten Artikel der Listen alle möglichen Kombinationen bekommen, aber nur wenn die ersten Elemente in beiden Listen gleich sind. Also, was gedruckt werden soll, ist der folgende:

Combinations for 'a' as first item in the lists: [b,f], [b,c], [f,c] 

ich kam mit diesem:

comprehension = [x[0] for x in nested_list if x[1] in nested_list] 

Aber das funktioniert nicht (natürlich). Ich bin mir nicht sicher, wie ich alle Listen durchlaufen soll, um die Kombinationen zu finden.

Answer 1

können Sie verwenden collectoins.defaultdict und combinations:

In [2]: nested_list = [['a','b',1],['c','d',3],['a','f',8],['a','c',5]] 

In [3]: from collections import defaultdict 

In [4]: from itertools import combinations 

In [5]: d = defaultdict(list)  

In [6]: for i, j, _ in nested_list: 
    ...:  d[i].append(j) 
    ...:  

In [7]: {k: list(combinations(v, 2)) for k, v in d.items()} 
Out[7]: {'a': [('b', 'f'), ('b', 'c'), ('f', 'c')], 'c': []}