Skip to main content

Collection of network-related utilities for python.

Project description

Build Status

Pelote

Pelote is a python library full of graph-related functions that can be used to complement networkx for higher-level tasks.

It mainly helps with the following things:

  • Conversion of tabular data to graphs (bipartites, citation etc. in the spirit of Table2Net)
  • Conversion of graphs to tabular data
  • Monopartite projections of bipartite graphs
  • Miscellaneous graph helper functions (filtering out nodes, edges etc.)
  • Sparsification of graphs
  • Reading & writing of graph formats not found in networkx (such as graphology JSON)

As such it is the perfect companion to ipysigma, our Jupyter widget that can render interactive graphs directly within your notebooks.

Installation

You can install pelote with pip with the following command:

pip install pelote

If you want to be able to use the library with pandas, you will need to install it also:

pip install pandas

Usage


Tabular data to graphs

table_to_bipartite_graph

Function creating a bipartite graph from the given tabular data.

Arguments

  • table Iterable[Indexable] or pd.DataFrame - input tabular data. It can be a large variety of things as long as it is 1. iterable and 2. yields indexable values such as dicts or lists. This can for instance be a list of dicts, a csv.DictReader stream etc. It also supports pandas DataFrame if the library is installed.
  • first_part_col Hashable - the name of the column containing the value representing a node in the resulting graph's first part. It could be the index if your rows are lists or a key if your rows are dicts instead.
  • second_par_col Hashable - the name of the column containing the value representing a node in the resulting graph's second part. It could be the index if your rows are lists or a key if your rows are dicts instead.
  • node_part_attr str, optional "part" - name of the node attribute containing the part it belongs to.
  • edge_weight_attr str, optional "weight" - name of the edge attribute containing its weight, i.e. the number of times it was found in the table.
  • first_part_data Sequence or Callable or Mapping, optional None - sequence (i.e. list, tuple etc.) of column from rows to keep as node attributes for the graph's first part. Can also be a mapping (i.e. dict) from row column to node attribute name to create. Can also be a function returning a dict of those attributes. Note that the first row containing a given node will take precedence over subsequent ones regarding data to include.
  • second_part_data Sequence or Callable or Mapping, optional None - sequence (i.e. list, tuple etc.) of column from rows to keep as node attributes for the graph's second part. Can also be a mapping (i.e. dict) from row column to node attribute name to create. Can also be a function returning a dict of those attributes. Note that the first row containing a given node will take precedence over subsequent ones regarding data to include.
  • first_part_name Hashable, optional None - can be given to rename the first part.
  • second_part_name Hashable, optional None - can be given to rename the second part. to display as graph's second part's name.
  • disjoint_keys bool, optional False - set this to True as an optimization mechanism if you know your part keys are disjoint, i.e. if no value for first_part_col can also be found in second_part_col. If you enable this option wrongly, the result can be incorrect.

Returns

nx.AnyGraph - the bipartite graph.

tables_to_graph

Function creating a graph from two tables: a table of nodes and a table of edges.

from pelote import tables_to_graph

table_nodes = [
    {"name": "alice", "age": 50},
    {"name": "bob", "age": 12}
]

table_edges = [
    {"source": "alice", "target": "bob", "weight": 0.8},
    {"source": "bob", "target": "alice", "weight": 0.2}
]

g = tables_to_graph(
    table_nodes, table_edges, node_col="name", node_data=["age"], edge_data=["weight"], directed=True
)

Arguments

  • nodes_table Iterable[Indexable] or pd.DataFrame - input nodes in tabular format. It can be a large variety of things as long as it is 1. iterable and 2. yields indexable values such as dicts or lists. This can for instance be a list of dicts, a csv.DictReader stream etc. It also supports pandas DataFrame if the library is installed.
  • edges_table Iterable[Indexable] or pd.DataFrame - input edges in tabular format.
  • node_col Hashable, optional "key" - the name of the column containing the nodes in the nodes_table. It could be the index if your rows are lists or a key if your rows are dicts instead.
  • edge_source_col Hashable, optional "source" - the name of the column containing the edges' source nodes in the edges_table.
  • edge_target_col Hashable, optional "target" - the name of the column containing the edges' target nodes in the edges_table.
  • node_data Sequence, optional [] - sequence (i.e. list, tuple etc.) of columns' names from the nodes_table to keep as node attributes in the resulting graph.
  • edge_data Sequence, optional [] - sequence (i.e. list, tuple etc.) of columns' names from the edges_table to keep as edge attributes in the resulting graph, e.g. ["weight"].
  • count_rows_as_weight bool, optional False - set this to True to compute a weight attribute for each edge, corresponding to the number of times it was found in the table. The name of this attribute is defined by the edge_weight_attr parameter. If set to False, only the last occurrence of each edge will be kept in the graph.
  • edge_weight_attr str, optional "weight" - name of the edge attribute containing its weight, i.e. the number of times it was found in the table, if count_rows_as_weight is set to True.
  • add_missing_nodes bool, optional True - set this to True to check that the edges' sources and targets in the edges_table are all defined in the nodes_table.
  • directed bool, optional False - whether the resulting graph must be directed.

Returns

nx.AnyGraph - the resulting graph.

edges_table_to_graph

Function creating a graph from a table of edges.

Arguments

  • edges_table Iterable[Indexable] or pd.DataFrame - input edges in tabular format. It can be a large variety of things as long as it is 1. iterable and 2. yields indexable values such as dicts or lists. This can for instance be a list of dicts, a csv.DictReader stream etc. It also supports pandas DataFrame if the library is installed.
  • edge_source_col Hashable, optional "source" - the name of the column containing the edges' source nodes in the edges_table.
  • edge_target_col Hashable, optional "target" - the name of the column containing the edges' target nodes in the edges_table.
  • edge_data Sequence, optional [] - sequence (i.e. list, tuple etc.) of columns' names from the edges_table to keep as edge attributes in the resulting graph, e.g. ["weight"].
  • count_rows_as_weight bool, optional False - set this to True to compute a weight attribute for each edge, corresponding to the number of times it was found in the table. The name of this attribute is defined by the edge_weight_attr parameter. If set to False, only the last occurrence of each edge will be kept in the graph.
  • edge_weight_attr str, optional "weight" - name of the edge attribute containing its weight, i.e. the number of times it was found in the table, if count_rows_as_weight is set to True.
  • directed bool, optional False - whether the resulting graph must be directed.

Returns

nx.AnyGraph - the resulting graph.


Graphs to tabular data

graph_to_nodes_dataframe

Function converting the given networkx graph into a pandas DataFrame of its nodes.

from pelote import graph_to_nodes_dataframe

df = graph_to_nodes_dataframe(graph)

Arguments

  • nx.AnyGraph - a networkx graph instance
  • node_key_col str, optional "key" - name of the DataFrame column containing the node keys. If None, the node keys will be used as the DataFrame index.

Returns

pd.DataFrame - A pandas DataFrame

graph_to_edges_dataframe

Function converting the given networkx graph into a pandas DataFrame of its edges.

Arguments

  • nx.AnyGraph - a networkx graph instance
  • edge_source_col str, optional "source" - name of the DataFrame column containing the edge source.
  • edge_target_col str, optional "target" - name of the DataFrame column containing the edge target.
  • source_node_data Iterable or Mapping, optional None - iterable of attribute names or mapping from attribute names to column name to be used to add columns to the resulting dataframe based on source node data.
  • target_node_data Iterable or Mapping, optional None - iterable of attribute names or mapping from attribute names to column name to be used to add columns to the resulting dataframe based on target node data.

Returns

pd.DataFrame - A pandas DataFrame

graph_to_dataframes

Function converting the given networkx graph into two pandas DataFrames: one for its nodes, one for its edges.

Arguments

  • nx.AnyGraph - a networkx graph instance
  • node_key_col str, optional "key" - name of the node DataFrame column containing the node keys. If None, the node keys will be used as the DataFrame index.
  • edge_source_col str, optional "source" - name of the edge DataFrame column containing the edge source.
  • edge_target_col str, optional "target" - name of the edge DataFrame column containing the edge target.
  • source_node_data Iterable or Mapping, optional None - iterable of attribute names or mapping from attribute names to column name to be used to add columns to the edge dataframe based on source node data.
  • target_node_data Iterable or Mapping, optional None - iterable of attribute names or mapping from attribute names to column name to be used to add columns to the edge dataframe based on target node data.

Returns

None - (pd.DataFrame, pd.DataFrame)


Graph projection

monopartite_projection

Function returning the monopartite projection of a given bipartite graph wrt one of both partitions of the graph.

That is to say the resulting graph will keep a single type of nodes sharing weighted edges based on the neighbors they shared in the bipartite graph.

import networkx as nx
from pelote import monopartite_projection

bipartite = nx.Graph()
bipartite.add_nodes_from([1, 2, 3], part='account')
bipartite.add_nodes_from([4, 5, 6], part='color')
bipartite.add_edges_from([
    (1, 4),
    (1, 5),
    (2, 6),
    (3, 4),
    (3, 6)
])

# Resulting graph will only contain nodes [1, 2, 3]
# with edges: (1, 3) and (2, 3)
monopartite = monopartite_projection(bipartite, 'account')

Arguments

  • bipartite_graph nx.AnyGraph - target graph. The function will raise if given graph is not truly bipartite.
  • part_to_keep Hashable or Collection - partition to keep in the projected graph. It can either be the value of the part node attribute in the given graph (a string, most commonly), or a collection (a set, list etc.) holding the nodes composing the part to keep.
  • node_part_attr str, optional "part" - name of the node attribute containing the part the node belongs to.
  • edge_weight_attr str, optional "weight" - name of the edge attribute containing the edge's weight.
  • metric str, optional None - one of "jaccard", "overlap", "cosine", "dice", "binary_cosine", "pmi" or "dot_product". If not given, resulting weight will be set to the size of neighbor intersection.
  • bipartition_check bool, optional True - whether to check if given graph is truly bipartite. You can disable this as an optimization strategy if you know what you are doing.
  • weight_threshold float, optional None - if an edge weight should be less than this threshold we would not add it to the projected monopartite graph.

Returns

nx.Graph - the projected monopartite graph.


Graph sparsification

global_threshold_sparsification

Function returning a copy of the given graph without edges whose weight is less than a given threshold.

Arguments

  • graph nx.AnyGraph - target graph.
  • weight_threshold float - weight threshold.
  • edge_weight_attr str, optional - name of the edge weight attribute.
  • reverse bool, optional - whether to reverse the threshold condition. That is to say an edge would be removed if its weight is greater than the threshold.
  • keep_connected bool, optional False - whether to keep the graph connected as it is using the UMST method.

Returns

nx.AnyGraph - the sparse graph.

multiscale_backbone

Function returning the multiscale backbone of the given graph, i.e. a copy of the graph were we only kept "relevant" edges, as defined by a statistical test where we compare the likelihood of a weighted edge existing vs. the null model.

Article

Serrano, M. Ángeles, Marián Boguná, and Alessandro Vespignani. "Extracting the multiscale backbone of complex weighted networks." Proceedings of the national academy of sciences 106.16 (2009): 6483-6488.

References

Arguments

  • graph nx.AnyGraph - target graph.
  • alpha float, optional 0.05 - alpha value for the statistical test. It can be intuitively thought of as a p-value score for an edge to be kept in the resulting graph.
  • edge_weight_attr str, optional "weight" - name of the edge attribute holding the edge's weight.
  • keep_connected bool, optional False - whether to keep the graph connected as it is using the UMST method.

Returns

nx.AnyGraph - the sparse graph.


Miscellaneous graph-related metrics

edge_disparity

Function computing the disparity score of each edge in the given graph. This score is typically used to extract the multiscale backbone of a weighted graph.

The formula from the paper (relying on integral calculus) can be simplified to become:

disparity(u, v) = min(
    (1 - normalizedWeight(u, v)) ^ (degree(u) - 1)),
    (1 - normalizedWeight(v, u)) ^ (degree(v) - 1))
)

where

normalizedWeight(u, v) = weight(u, v) / weightedDegree(u)
weightedDegree(u) = sum(weight(u, v) for v in neighbors(u))

This score can sometimes be found reversed likewise:

disparity(u, v) = max(
    1 - (1 - normalizedWeight(u, v)) ^ (degree(u) - 1)),
    1 - (1 - normalizedWeight(v, u)) ^ (degree(v) - 1))
)

so that higher score means better edges. We chose to keep the metric close to the paper to keep the statistical test angle. This means that, in this implementation at least, a low score for an edge means a high relevance and increases its chances to be kept in the backbone.

Note that this algorithm has no proper definition for directed graphs and is only useful if edges have varying weights. This said, it could be possible to compute the disparity score only based on edge direction, if we drop the min part.

Article

Serrano, M. Ángeles, Marián Boguná, and Alessandro Vespignani. "Extracting the multiscale backbone of complex weighted networks." Proceedings of the national academy of sciences 106.16 (2009): 6483-6488.

References

Arguments

  • graph nx.AnyGraph - target graph.
  • edge_weight_attr str, optional "weight" - name of the edge attribute containing its weight.
  • reverse bool, optional False - whether to reverse the metric, i.e. higher weight means more relevant edges.

Returns

dict - Dictionnary with edges - (source, target) tuples - as keys and the disparity scores as values

triangular_strength

Function returning a graph edges triangular strength, sometimes also called Simmelian strength, i.e. the number of triangles each edge is a part of.

Arguments

  • graph nx.AnyGraph - target graph.
  • full bool, optional False - whether to return strength for every edge, including those with strength = 0.

Returns

dict - mapping of edges to their triangular strength.


Graph utilities

union_of_maximum_spanning_trees

Generator yielding the edges belonging to any Maximum Spanning Tree (MST) of the given networkx graph.

Note that this function will give to each edge with no weight a default weight of 1.

Article

Arlind Nocaj, Mark Ortmann, and Ulrik Brandes "Untangling Hairballs. From 3 to 14 Degrees of Separation." Computer & Information Science, University of Konstanz, Germany, 2014, https://dx.doi.org/10.1007/978-3-662-45803-7_9.

References

Arguments

  • graph nx.AnyGraph - target graph.
  • edge_weight_attr str, optional "weight" - name of the edge weight attribute.

Yields

tuple - source, target, attributes

largest_connected_component

Function returning the largest connected component of given networkx graph as a set of nodes.

Note that this function will consider any given graph as undirected and will therefore work with weakly connected components in the directed case.

Arguments

  • graph nx.AnyGraph - target graph.

Returns

set - set of nodes representing the largest connected component.

crop_to_largest_connected_component

Function mutating the given networkx graph in order to keep only the largest connected component.

Note that this function will consider any given graph as undirected and will therefore work with weakly connected components in the directed case.

Arguments

  • graph nx.AnyGraph - target graph.

largest_connected_component_subgraph

Function returning the largest connected component subgraph of the given networkx graph.

Note that this function will consider any given graph as undirected and will therefore work with weakly connected components in the directed case.

Arguments

  • graph nx.AnyGraph - target graph.
  • as_view bool, optional False - whether to return the subgraph as a view.

Returns

nx.AnyGraph - the subgraph.

remove_edges

Function removing all edges that do not pass a predicate function from a given networkx graph.

Note that this function mutates the given graph.

Arguments

  • graph nx.AnyGraph - a networkx graph.
  • predicate callable - a function taking each edge source, target and attributes and returning True if you want to keep the edge or False if you want to remove it.

filter_edges

Function returning a copy of the given networkx graph but without the edges filtered out by the given predicate function

Arguments

  • graph nx.AnyGraph - a networkx graph.
  • predicate callable - a function taking each edge source, target and attributes and returning True if you want to keep the edge or False if you want to remove it.

Returns

nx.AnyGraph - the filtered graph.

remove_nodes

Function removing all nodes that do not pass a predicate function from a given networkx graph.

Note that this function mutates the given graph.

from pelote import remove_nodes

g = nx.Graph()
g.add_node(1, weight=22)
g.add_node(2, weight=4)
g.add_edge(1, 2)

remove_nodes(g, lambda n, a: a["weight"] >= 10)

Arguments

  • graph nx.AnyGraph - a networkx graph.
  • predicate callable - a function taking each node and node attributes and returning True if you want to keep the node or False if you want to remove it.

filter_nodes

Function returning a copy of the given networkx graph but without the nodes filtered out by the given predicate function

from pelote import filter_nodes

g = nx.Graph()
g.add_node(1, weight=22)
g.add_node(2, weight=4)
g.add_edge(1, 2)

h = filter_nodes(g, lambda n, a: a["weight"] >= 10)

Arguments

  • graph nx.AnyGraph - a networkx graph.
  • predicate callable - a function taking each node and node attributes and returning True if you want to keep the node or False if you want to remove it.

Returns

nx.AnyGraph - the filtered graph.

remove_leaves

Function removing all leaves of the graph, i.e. the nodes incident to a single edge, i.e. the nodes with degree 1.

This function is not recursive and will only remove one layer of leaves.

Note that this function mutates the given graph.

from pelote import remove_leaves

g = nx.Graph()
g.add_edge(1, 2)
g.add_edge(2, 3)

remove_leaves(g)

list(g.nodes)
>>> [2]

Arguments

  • graph nx.AnyGraph - a networkx graph.

filter_leaves

Function returning a copy of the given networkx graph but without its leaves, i.e. the nodes incident to a single edge, i.e. the nodes with degree 1.

This function is not recursive and will only filter only one layer of leaves.

from pelote import remove_leaves

g = nx.Graph()
g.add_edge(1, 2)
g.add_edge(2, 3)

h = filter_leaves(g)

list(h.nodes)
>>> [2]

Arguments

  • graph nx.AnyGraph - a networkx graph.

Learning

floatsam_threshold_learner

Function using an iterative algorithm to try and find the best weight threshold to apply to trim the given graph's edges while keeping the underlying community structure.

It works by iteratively increasing the threshold and stopping as soon as a significant connected component starts to drift away from the principal one.

This is basically an optimization algorithm applied to a complex nonlinear function using a very naive cost heuristic, but it works decently for typical cases as it emulates the method used by hand by some researchers when they perform this kind of task on Gephi, for instance.

When working on metrics where lower is better (i.e. edge disparity), you can reverse the logic of the algorithm by tweaking starting_threshold and giving a negative learning_rate.

Arguments

  • graph nx.Graph - Graph to sparsify.
  • starting_threshold float, optional 0.0 - Starting similarity threshold.
  • learning_rate float, optional 0.05 - How much to increase the threshold at each step of the algorithm.
  • max_drifter_order int, optional - Max order of component to detach itself from the principal one before stopping the algorithm. If not provided it will default to the logarithm of the graph's largest connected component's order.
  • edge_weight_attr str, optional "weight" - Name of the weight attribute.
  • on_epoch callable, optional - Function called on each epoch of the algorithm with some metadata about iteration state.

Returns

float - The found threshold


Reading & Writing

read_graphology_json

Function reading and parsing the given json file representing a serialized graphology graph as a networkx graph.

Note that this function cannot parse a true mixed graph since this is not supported by networkx.

Arguments

  • target str or Path or file or dict - target to read and parse. Can be a string path, a Path instance, a file buffer or already parsed JSON data as a dict.

Returns

nx.AnyGraph - a networkx graph instance.

write_graphology_json

Function serializing the given networkx graph as JSON, using the graphology format.

Note that both node keys and attribute names will be cast to string so they can safely be represented in JSON. As such in some cases (where your node keys and/or attribute names are not strings), this function will not be bijective when used with read_graphology_json.

Arguments

  • graph nx.AnyGraph - graph to serialize.
  • allow_mixed_keys bool, optional False - whether to allow graph with mixed node key types to be serialized nonetheless. Keys will always be cast to string so keys might clash and produce an invalid serialization. Only use this if you know what you are doing.
  • allow_invalid_attr_names bool, optional False - whether to allow non-string attribute names. Note that if you chose to allow them, some might clash and produce an invalid serialization. Only use this if you know what you are doing.

Returns

dict - JSON data

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pelote-0.8.2.tar.gz (38.6 kB view hashes)

Uploaded Source

Built Distribution

pelote-0.8.2-py3-none-any.whl (41.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page