Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def unir_vertices(grafo, arista, vertices):
- largo = len(vertices)
- for i in range(largo-1):
- vert1 = vertices[i]
- for j in range(i+1, largo):
- vert2 = vertices[j]
- grafo.agregar_arista(arista,vert1,vert2)
- def grafo_crear(nombre_archivo):
- """
- Crea un grafo de conexiones de actores a partir de un archivo de datos.
- PRE: Recibe el nombre de un archivo separado por comas que contenga de lineas:
- actor,pelicula,pelicula,pelicula
- que equivalen a: vertice,arista,arista,arista
- POST: Devuelve un grafo creado a partir de estos datos.
- """
- grafo = Grafo()
- actor_peli = {}
- with open(nombre_archivo, 'r') as f:
- archivo_csv = csv.reader(f)
- for linea in archivo_csv:
- actor = linea[0]
- grafo.agregar_vertice(actor)
- for i in range(1, len(linea)):
- peli = linea[i]
- if(peli not in actor_peli):
- actor_peli[peli] = [actor]
- else:
- actor_peli[peli].append(actor)
- for peli in actor_peli:
- unir_vertices(grafo, peli, actor_peli[peli])
- return grafo
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement