Advertisement
Guest User

Untitled

a guest
Dec 14th, 2017
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.01 KB | None | 0 0
  1.  
  2. def unir_vertices(grafo, arista, vertices):
  3.  
  4. largo = len(vertices)
  5.  
  6. for i in range(largo-1):
  7. vert1 = vertices[i]
  8. for j in range(i+1, largo):
  9. vert2 = vertices[j]
  10. grafo.agregar_arista(arista,vert1,vert2)
  11.  
  12. def grafo_crear(nombre_archivo):
  13. """
  14. Crea un grafo de conexiones de actores a partir de un archivo de datos.
  15. PRE: Recibe el nombre de un archivo separado por comas que contenga de lineas:
  16. actor,pelicula,pelicula,pelicula
  17. que equivalen a: vertice,arista,arista,arista
  18. POST: Devuelve un grafo creado a partir de estos datos.
  19. """
  20. grafo = Grafo()
  21. actor_peli = {}
  22.  
  23. with open(nombre_archivo, 'r') as f:
  24. archivo_csv = csv.reader(f)
  25. for linea in archivo_csv:
  26. actor = linea[0]
  27. grafo.agregar_vertice(actor)
  28. for i in range(1, len(linea)):
  29. peli = linea[i]
  30. if(peli not in actor_peli):
  31. actor_peli[peli] = [actor]
  32. else:
  33. actor_peli[peli].append(actor)
  34.  
  35. for peli in actor_peli:
  36. unir_vertices(grafo, peli, actor_peli[peli])
  37.  
  38. return grafo
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement