.. _sphx_glr_gallery_node_picking.py:


Interactively highlight nodes and edges
---------------------------------------

Run this with an interactive matplotlib backend!

Clicking on a node will hi-light it and it's edges



.. code-block:: python

    import networkx as nx
    import matplotlib.pyplot as plt
    from grave import plot_network
    from grave.style import use_attributes


    def hilighter(event):
        # if we did not hit a node, bail
        if not hasattr(event, 'nodes') or not event.nodes:
            return

        # pull out the graph,
        graph = event.artist.graph

        # clear any non-default color on nodes
        for node, attributes in graph.nodes.data():
            attributes.pop('color', None)

        for u, v, attributes in graph.edges.data():
            attributes.pop('width', None)

        for node in event.nodes:
            graph.nodes[node]['color'] = 'C1'

            for edge_attribute in graph[node].values():
                edge_attribute['width'] = 3

        # update the screen
        event.artist.stale = True
        event.artist.figure.canvas.draw_idle()


    graph = nx.barbell_graph(10, 14)

    fig, ax = plt.subplots()
    art = plot_network(graph, ax=ax, node_style=use_attributes(),
                       edge_style=use_attributes())

    art.set_picker(10)
    ax.set_title('Click on the nodes!')
    fig.canvas.mpl_connect('pick_event', hilighter)
    plt.show()

**Total running time of the script:** ( 0 minutes  0.000 seconds)



.. only :: html

 .. container:: sphx-glr-footer


  .. container:: sphx-glr-download

     :download:`Download Python source code: node_picking.py <node_picking.py>`



  .. container:: sphx-glr-download

     :download:`Download Jupyter notebook: node_picking.ipynb <node_picking.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.readthedocs.io>`_