Das deutsche Python-Forum

Code: Alles auswählen

def make_colormap(visited, visiting):
    color_map = list()
    for boolean in visited:
        if boolean:
            color_map.append('grey')
        else:
            color_map.append('black')
    color_map[visiting-1]= 'red'
    return color_map



def tiefensuche(start, visited, g=nx.Graph()):
    if len(visited)<2:
        print("Graph too small")
    pos = nx.spring_layout(g)
    visiting = start

    visited[visiting - 1] = True
    for neighbour in g.neighbors(visiting):
        if not visited[neighbour-1]:
            color_map=make_colormap(visited, visiting)
            #hier
            tiefensuche(neighbour, visited, g)
   
    # fehlt ein teil wollte net zu lange machen        
  pos = nx.spring_layout(g)
    nx.draw_networkx_nodes(G, pos)
    nx.draw_networkx_edges(G, pos, edgelist=G.edges(), width=6)
    nx.draw_networkx_labels(G, pos)