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