networkxのノードの色分けについて

Question

networkxでグラフを作るにあたってノードの色を条件で分けたいです
現在つぎのようなコードを組みましたがエラーが出てしまいます

import matplotlib.pyplot as plt
import networkx as nx

 G = nx.Graph() 
 for i in range(100):
   if i == 0:
     G.add_node(i)
   elif i%3 == 0 and i%2 == 0:
     G.add_node(i, color = "blue")
     G.add_edge(0,i)
   elif i%3 == 0:
     G.add_node(i, color = "green")
     G.add_edge(0,i)
   elif i%2 == 0:
     G.add_node(i, color = "red")
     G.add_edge(0,i)

pos = nx.spring_layout(G, k=1)
colors = nx.get_node_attributes(G,'color').values()
nx.draw_networkx(G, pos,node_color = colors )
plt.show()

このコードを実行したところ
ValueError: 'c' argument has 24 elements, which is inconsistent with 'x' and 'y' with size 25.
というエラーがでました
このようにエッジの色わけをするにはどのようなコードを組めばよいでしょうか

Answer 1

i == 0 の場合のカラー指定が足りません。

G = nx.Graph() 
for i in range(100):
  if i == 0:
    G.add_node(i, color = "gold")
  elif i%3 == 0 and i%2 == 0:
              :