msticpy.transform.network module

Module for converting DataFrame to Networkx graph.

msticpy.transform.network.df_to_networkx(data: DataFrame, source_col: str, target_col: str, source_attrs: Optional[Iterable[str]] = None, target_attrs: Optional[Iterable[str]] = None, edge_attrs: Optional[Iterable[str]] = None, graph_type: Literal['graph', 'digraph'] = 'graph')

Create a networkx graph from a DataFrame.

Parameters

• data (pd.DataFrame) – Input data

• source_col (str) – Column for source nodes.

• target_col (str) – Column for target nodes.

• source_attrs (Optional[List[str]], optional) – Optional list of columns to use as source node attributes, by default None

• target_attrs (Optional[List[str]], optional) – Optional list of columns to use as target node attributes, by default None

• edge_attrs (Optional[List[str]], optional) – Optional list of columns to use as edge node attributes, by default None

• graph_type (str) – “graph” or “digraph” (for nx.DiGraph)

Returns

The networkx graph object

Return type

nx.Graph