def from_pandas_adjacency(df, create_using=Graph):
"""
Initializes the graph from pandas adjacency matrix.
Set create_using to cugraph.DiGraph for directed graph and
cugraph.Graph for undirected Graph.
"""
if create_using is Graph:
G = Graph()
elif create_using is DiGraph:
G = DiGraph()
else:
raise Exception("create_using supports Graph and DiGraph")
G.from_pandas_adjacency(df)
return G
def from_pandas_adjacency(df, create_using=Graph):
"""
Initializes the graph from pandas adjacency matrix.
Parameters
----------
df : pandas.DataFrame
A DataFrame that contains edge information
create_using: cugraph.DiGraph or cugraph.Graph, optional (default=Graph)
Indicate whether to create a directed or undirected graph
"""
if create_using is Graph:
G = Graph()
elif create_using is DiGraph:
G = DiGraph()
else:
raise Exception("create_using supports Graph and DiGraph")
G.from_pandas_adjacency(df)
return G