Networkx неправильно окрашивает края по атрибуту

Я пытаюсь изобразить транспортную сеть в networkx, и до сих пор у меня это хорошо получалось, но проблема в следующем: Я загрузил список краев из текстового файла, который имеет следующую форму:

61  1
54  79
66  134
68  57

и т. д. Кроме того, чтобы отобразить узлы сети в правильном положении, я загрузил и добавил в график txt с координатами узлов, который имеет следующий вид:

1   478947.434  4204495.502
2   478909.145  4204244.629
3   479065.936  4204709.003
4   478880.827  4204297.676
5   478993.409  4204167.957

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

import networkx as nx
import numpy as py
import copy
import matplotlib.pyplot as plt
import xlwings as xw
from xlwings import Book, Range

pos = {}
with open(r'C:\Users\alexl\Documents\new_python\nodes_coordinates.txt') as f:
    for line in f:
        node, x, y = line.split()
        pos[node] = float(x), float(y)

print(pos)
network0 = nx.read_edgelist(r'C:\Users\alexl\Documents\new_python\init_edgelist.txt', create_using = nx.OrderedDiGraph)

nx.draw(network0, pos = pos, with_labels = True, edge_color = G, edge_cmap = plt.cm.Reds, connectionstyle = 'arc3,rad=0.1')
         

plt.show()

Теперь я хочу правильно назначить «веса» каждому ребру и раскрасить его по этому атрибуту. Для каждого ребра есть 3 возможных результата: 0, 1 и 2, поэтому я хочу 3 разных цвета, например, красный для 0, зеленый для 1 и синий для 2. У меня есть словарь Python, подобный этому:

{0: 1.0, 1: 1.0, 2: 0.0, 3: 1.0, 4: 1.0, 5: 0.0, 6: 0.0, 7: 0.0, 8: 1.0,...

и т. д. Первое значение (1.0) соответствует первому краю списка выше, второе (1.0) — второму и т. д. Но я вижу, что networkx этого не понимает и неправильно раскрашивает мою сеть. Я преобразовал словарь в список и сделал следующее:

var1 = G.values()
list1 = list(var1)
col = []
for n in range(len(list1)):
    if list1[n] == 0:
        col.append('b')
    elif list1[n] == 1:
        #col.append('r')
    else:
        #col.append('g') 
nx.draw(network0, pos = pos, with_labels = True, edge_color = col, edge_cmap = plt.cm.Reds, connectionstyle = 'arc3,rad=0.1')

Я проверил результат, но соответствие не верное. Я думаю, что если в списке 32 '0, он окрашивает 32 ребра тем цветом, который я ему назначил, но не правильными. (Но каждый раз, когда я запускаю его, одни и те же 32 ребра окрашиваются таким образом, так что, вероятно, это не случайно.) Интересно, что не так. Я пробовал его также как словарь (не преобразовывая в список), но на самом деле цветов больше 3, я не знаю, почему. Должен ли я попытаться добавить эти значения (0, 1, 1, 2 и т. д.) в edgelist.txt, чтобы назначение было выполнено правильно?