python - durch die Liste iterieren, um das aktuelle Element als Schlüssel, und die vorderen und hinteren Elemente als Werte zuzuordnen?

Question

alist = [ 'A', 'B ' C', 'D']

was diese Liste repräsentiert ist: A < -> B < -> C < -> D

Ich möchte auf Karte wie die folgenden:

A: B

B: A, C

C: B, D

D: C

Zuerst habe ich:

Aber das ist offensichtlich falsch, weil es zwei Elemente überspringt.

alist = ['A','B', 'C', 'D'] 
graph = {} 
i = 2 
while i < len(alist): 
    current, front, back = alist[i-1], alist[i-2], alist[i] 
    if current not in graph: 
     graph[current] = [front, back] 
    else: 
     graph[current].append(front) 
     graph[current].append(back) 
    i+=1 

Wie könnte ich das beheben? Bare mit mir bitte, ich bin wirklich schlecht mit Iterationen ..

Answer 1

Schon jetzt gibt es viele Lösungen, das ist ein sehr einfach zu verstehen:

alist = ['A','B', 'C', 'D'] 
graph = {} 
for index, item in enumerate(alist): 
    if index == 0: 
     if len(alist) == 1: 
      graph[item] = [] 
      break 
     graph[item] = [alist[index+1]] 
    elif index == len(alist)-1: 
     graph[item] = [alist[index-1]] 
    else: 
     graph[item] = [alist[index-1], alist[index+1]] 

Ausgang:

>>> print graph 

154: {'A': ['B'], 'B': ['A', 'C'], 'C': ['B', 'D'], 'D': ['C']} 

Dank @mglison für die Ausnahme unter Hinweis darauf, wenn len(alist) == 1, herausgegeben

Answer 2

Was ist so etwas wie:

graph = {} 
for i, item in enumerate(alist): 
    connections = [] 
    for j in (i-1, i+1): # look at element before and after the i'th 
     if 0 <= j < len(alist): # check if the index is valid for `alist` 
      connections.append(alist[j]) 
    graph[item] = connections 

Beachten Sie, dass dies nur funktioniert, für alist ist eine Folge von irgendeiner Art. Es wird nicht für Generatoren funktionieren. Außerdem bin ich normalerweise kein Fan der Überprüfung, ob ein Index in der Liste ist. Ich würde eher try: ... except IndexError: ... (EAFP), aber IndexError würde nicht für negative j ausgelöst werden, so dass ich diese Route für diese bestimmte Anwendung vermieden).

Answer 3

Dies ist fast definitiv nicht das, was Sie wollen, aber es war so eine süße One-Line-Lösung für Ihr Problem, dem ich nicht widerstehen konnte.

> l = ['A','B','C','D'] 
> { k: filter(None, [v1,v2]) for (k,v1,v2) in zip(l,[None]+l[:-1],l[1:]+[None]) } 
{'A': ['B'], 'C': ['B', 'D'], 'B': ['A', 'C'], 'D': ['C']} 

Answer 4

Hier ist eine schnelle und schmutzige Lösung (es gibt bessere Möglichkeiten, die ich für längere Gegenstände sicher bin).

graph = {alist[0]: alist[1], alist[-1]: alist[-2]} # terminal elements 
for i in range(1, len(alist)-1):     # middle elements 
    graph[alist[i]] = [alist[i-1], alist[i+1]] 

In [43]: for k, v in sorted(graph.iteritems()): 
    print "{} -> {}".format(k, v) 
    ....:  
A -> B 
B -> ['A', 'C'] 
C -> ['B', 'D'] 
D -> C