edgegraph.traversal.helpers.neighbors

Contents

edgegraph.traversal.helpers.neighbors#

edgegraph.traversal.helpers.neighbors(vert, direction_sensitive=True, unknown_handling=2, filterfunc=None)#

Identify the neighbors of a given vertex.

This function checks the edges associated with the given vertex, identifies the vertices on the other end of those edges, and returns them. It respects edge directionality if/when necessary, and can handle arbitrary edge types given they are subclasses of either DirectedEdge or UnDirectedEdge.

For example, with the given graph:

object v1 object v2 object v3 object v4 v1 --> v2 v1 --> v3 v2 --> v3 v3 --> v4 v4 --> v1

the function would operate as:

>>> neighbors(v1)
[v2, v3]
>>> neighbors(v1, direction_sensitive=False)
[v2, v3, v4]
>>> neighbors(v4)
[v1]
>>> neighbors(v4, direction_sensitive=False)
[v1, v3]

If supplied, the filterfunc argument should be to a callable object (function or otherwise) that will return either True or False. This function is used to determine if a given vertex should be included in the returned neighbors. It must have the following signature:

edgegraph.traversal.helpers.filterfunc(e, v2)

Determines if a given vertex (v2) should be included in the neighbors of v. Because v2 may be reachable from v1 via multiple edges, the edge currently being considered is given as well.

Parameters:

  • e – The edge connecting v1 to v2.

  • v2 – The vertex under consideration.

Returns:

Whether or not v2 should be considered a neighbor of v, when reached via e.

For example, one may wish to only consider vertices if a given attribute meets some criteria:

>>> neighbors(v1)
[v2, v3]
>>> neighbors(v1, filterfunc=lambda e, v2: v2.i >= 3)
[v3]

Note

The filterfunc parameter operates in addition to the direction_sensitive parameter!

Parameters:

  • vert (Vertex) – The vertex to identify neighbors of.

  • direction_sensitive (bool) – Enables handling of directional links. If set to False, any subclasses / instances of DirectedEdge are treated as if they were instead subclasses / instances of UnDirectedEdge.

  • unknown_handling (int) – What to do with edges whose class is not recognized (not a subclass / instance of either DirectedEdge or UnDirectedEdge). Options are LNK_UNKNOWN_NONNEIGHBOR to treat unknown edges as non-neighbors, LNK_UNKNOWN_NEIGHBOR to treat unknown edges as neighbors, or LNK_UNKNOWN_ERROR to raise a NotImplementedError if such an edge is encountered.

  • filterfunc (Callable) – Callable object to determine whether the given edge / vertex should be included in the neighbors output.

Raises:

NotImplementedError – if kwarg unknown_handling is set to LNK_UNKNOWN_ERROR and an unknown edge class is enountered.

Returns:

A list of Vertex objects representing neighbors of the specified vertex.

Return type:

list[Vertex]

Contents