File size: 2,339 Bytes
e1a22ca 0cbb473 e1a22ca c09e84e e1a22ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import pandas as pd
import networkx as nx
import matplotlib.pyplot as plt
from pyvis.network import Network
from data import df
def analysis():
G = nx.DiGraph()
# Define the columns to consider for relationships
relationship_columns = ['father', 'sibling', 'spouse', 'mother', 'child']
G.add_nodes_from(df["itemLabel"])
for index, row in df.iterrows():
main_entity = row['itemLabel']
for relationship in relationship_columns:
if pd.notna(row[relationship]):
G.add_edge(main_entity, row[relationship], relationship=str(relationship), label=relationship)
plt.figure(figsize=(50, 20)) # Set the size of the plot
# pos = nx.kamada_kawai_layout(G) # Compute the positions of the nodes
# # Draw the nodes and edges with labels
# nx.draw(G, pos, with_labels=True, node_size=20, node_color='skyblue', font_size=10, font_color='black', font_weight='bold', alpha=0.7)
# # Draw edge labels (the relationships)
# edge_labels = nx.get_edge_attributes(G, 'relationship')
# nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_color='red')
# plt.title('Relationship Graph with Labels')
# plt.axis('off') # Hide the axes for clarity
# plt.show()
net = Network(width="1200px", height="800px", bgcolor="#222222", font_color="white", directed=True)
net.from_nx(G)
for node in net.nodes:
node["title"] = node["id"]
node["value"] = len(G[node["id"]])
for edge in net.edges:
# Set edge title to the relationship from your graph
relationship = G[edge["from"]][edge["to"]].get("relationship", "Unknown")
edge["title"] = relationship # This will show as a tooltip
edge["color"] = "blue"
edge["width"] = 2 if relationship != "Unknown" else 1
html = net.generate_html()
#need to remove ' from HTML
html = html.replace("'", "\"")
return f"""<iframe style="width: 100%; height: 800px;margin:0 auto" name="result" allow="midi; geolocation; microphone; camera;
display-capture; encrypted-media;" sandbox="allow-modals allow-forms
allow-scripts allow-same-origin allow-popups
allow-top-navigation-by-user-activation allow-downloads" allowfullscreen=""
allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""
|