Organize files
move some test files from root to example
This commit is contained in:
40
examples/get_all_edges_nx.py
Normal file
40
examples/get_all_edges_nx.py
Normal file
@@ -0,0 +1,40 @@
|
||||
import networkx as nx
|
||||
|
||||
G = nx.read_graphml("./dickensTestEmbedcall/graph_chunk_entity_relation.graphml")
|
||||
|
||||
|
||||
def get_all_edges_and_nodes(G):
|
||||
# Get all edges and their properties
|
||||
edges_with_properties = []
|
||||
for u, v, data in G.edges(data=True):
|
||||
edges_with_properties.append(
|
||||
{
|
||||
"start": u,
|
||||
"end": v,
|
||||
"label": data.get(
|
||||
"label", ""
|
||||
), # Assuming 'label' is used for edge type
|
||||
"properties": data,
|
||||
"start_node_properties": G.nodes[u],
|
||||
"end_node_properties": G.nodes[v],
|
||||
}
|
||||
)
|
||||
|
||||
return edges_with_properties
|
||||
|
||||
|
||||
# Example usage
|
||||
if __name__ == "__main__":
|
||||
# Assume G is your NetworkX graph loaded from Neo4j
|
||||
|
||||
all_edges = get_all_edges_and_nodes(G)
|
||||
|
||||
# Print all edges and node properties
|
||||
for edge in all_edges:
|
||||
print(f"Edge Label: {edge['label']}")
|
||||
print(f"Edge Properties: {edge['properties']}")
|
||||
print(f"Start Node: {edge['start']}")
|
||||
print(f"Start Node Properties: {edge['start_node_properties']}")
|
||||
print(f"End Node: {edge['end']}")
|
||||
print(f"End Node Properties: {edge['end_node_properties']}")
|
||||
print("---")
|
Reference in New Issue
Block a user