Python для итерации цикла

Question

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

sec=[ 
    [1, 2, 4.0, 100.0], 
    [2, 3, 1.5, 500.0], 
    [2, 4, 10.0, 700.0], 
    [2, 5, 5.75, 1500.0], 
    [3, 4, 11.4, 200.0], 
    [4, 5, 10.5, 750.0], 
    [4, 6, 6.75, 550.0]] 

Я сделал этот список и есть также этот словарь

graph={1: [1], 2: [2], 3: [3], 4: [4], 5: [5], 6: [6]} , что я пытаюсь сделать это

graph = {1: ['2'], 
     2: ['3', '4','5'], 
     3: ['4'], 
     4: ['5','6'], 
     5: [], 
     6: []} 

как он должен приста k состоит в том, что он собирает все sec[x][0] в качестве ключа словаря и sec[n][1] как значения в словарях , если sec [x] [0] имеет значение 1 и sec [x] [1] имеет значение 2, тогда число 2 равно добавляется в словарь в качестве значения для ключа 1 код, который я получил это

def magic(numList): #turns string into int 
    s = ''.join(map(str, numList)) 
    return int(s) 
for i in range(1,len(loc)+1): #len of loc is 6 
    for n in range(0,len(loc)+1): 
     if magic(graph[i])==sec[n][0]: 
      graph[i].append(sec[n][1]) 

, но это будет только добавить первое значение, то индекс п остановит отсчет, а затем индекс я продолжает идти и поэтому он не будет проверять больше значений в ключах

Answer 1

Ваше первоначальное определение graph не подходит для предполагаемого результата. Инициализируйте значения пустыми списками, затем добавьте их в простой цикл:

>>> sec=[ 
    [1, 2, 4.0, 100.0], 
    [2, 3, 1.5, 500.0], 
    [2, 4, 10.0, 700.0], 
    [2, 5, 5.75, 1500.0], 
    [3, 4, 11.4, 200.0], 
    [4, 5, 10.5, 750.0], 
    [4, 6, 6.75, 550.0]] 
>>> graph = {i:[] for i in range(1,7)} 
>>> for x,y,*z in sec: graph[x].append(str(y)) 

>>> graph 
{1: ['2'], 2: ['3', '4', '5'], 3: ['4'], 4: ['5', '6'], 5: [], 6: []} 

Answer 2

Используйте defaultdict из модуля коллекций. Смотрите ниже:

from collections import defaultdict 

graph_dd = defaultdict(list) 

for s in sec: 
    from_, to_ = s[:2] 
    graph_dd[from_].append(to_) 
    # do-nothing reference to to_ creates empty list, so that all nodes are represented 
    # in graph_dd; else 5 and 6 would not be created 
    graph_dd[to_] 

# convert to regular dict (not really necessary, but OP wanted a dict) 
graph = dict(graph_dd.items()) 

for k,v in graph.items(): 
    print(k,':', v) 

печатает:

1 : [2] 
2 : [3, 4, 5] 
3 : [4] 
4 : [5, 6] 
5 : [] 
6 : []