You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

76 lines
2.3 KiB
Python

3 years ago
#!/usr/bin/env python3
from itertools import combinations
from math import comb
from matplotlib.pyplot import title
import networkx as nx
from pyvis.network import Network
net = Network(height='100%', width='100%', directed=False, bgcolor='#1e1f29', font_color='white')
actors_to_keep = []
harmonic_to_keep =[]
3 years ago
with open('data/top_actors_h.txt') as ifs:
for line in ifs:
if line.strip():
actor_id, harmonic = line.split(maxsplit=1)
3 years ago
actors_to_keep.append(int(actor_id))
harmonic_to_keep.append(float(harmonic))
3 years ago
with open('data/Attori.txt') as ifs:
for line in ifs:
if line.strip():
actor_id, actor_name = line.split(maxsplit=1)
actor_id = int(actor_id)
harmonic = float(harmonic)
3 years ago
if actor_id in actors_to_keep:
if harmonic in harmonic_to_keep:
net.add_node(actor_id, label=actor_name, size = harmonic/350)
3 years ago
movies = {} # {movie_id: [actor_id, ...]}
with open('data/Relazioni.txt') as ifs:
for line in ifs:
if line.strip():
movie_id, actor_id = line.split(maxsplit=1)
actor_id = int(actor_id)
movie_id = int(movie_id)
if actor_id not in net.node_ids:
continue
if movie_id in movies:
movies[movie_id].append(actor_id)
else:
movies[movie_id] = [actor_id]
edges = set() # set of unique tuples (actor_id, actor_id)
for movie_id, actors in movies.items():
actors.sort()
for actor_id_1, actor_id_2 in combinations(actors, 2):
edges.add((actor_id_1, actor_id_2))
for actor_id_1, actor_id_2 in edges:
net.add_edge(actor_id_1, actor_id_2)
# net.hrepulsion(node_distance=500, central_gravity=0.3, spring_length=500, spring_strength=0.05, damping=0.2)
# net.repulsion(node_distance=500, central_gravity=0.3, spring_length=200, spring_strength=0.05, damping=0.2)
# net.show_buttons()
net.set_options("""
var options = {
"edges": {
"color": {
"inherit": true
},
"smooth": false
},
"physics": {
"repulsion": {
"springLength": 1205,
"nodeDistance": 1190
},
"maxVelocity": 23,
"minVelocity": 0.75,
"solver": "repulsion"
}
}
""")
net.show('html-files/harmonic-graph.html')