Visualisation Python

Écrire du code Python pour visualiser un graphique dirigé pondéré avec des exemples de données

Vous pouvez interroger un modèle directement et tester les résultats renvoyés lors de l'utilisation de différentes valeurs de paramètre en utilisant la console Cloud ou en appelant directement l'API Vertex AI.

Requête forme libre

Écrire du code Python pour dessiner un graphique orienté à l'aide de matplotlib et de networkx Montrez comment utiliser le code avec des exemples de données. Incluez toutes les instructions d'importation nécessaires.

Réponse



import matplotlib.pyplot as plt
import networkx as nx

# Example data: a directed graph with nodes and edges
nodes = ['A', 'B', 'C', 'D', 'E']
edges = [('A', 'B'), ('A', 'C'), ('B', 'D'), ('C', 'E'), ('D', 'E')]

# Create a directed graph object
graph = nx.DiGraph()

# Add nodes to the graph
graph.add_nodes_from(nodes)

# Add edges to the graph
graph.add_edges_from(edges)

# Draw the graph using matplotlib
nx.draw(graph, with_labels=True, node_color='lightblue', node_size=500, font_size=10, arrowsize=20)

# Display the graph
plt.show()

Explication :

  1. Importer les bibliothèques nécessaires :
  2. matplotlib.pyplot pour tracer le graphique

  3. networkx pour créer et manipuler le graphique

  4. Définir des exemples de données :

  5. nodes : liste de noms de nœuds.

  6. edges : liste de tuples représentant les arêtes dirigées, où se trouve chaque tuple (nœud source, nœud cible)

  7. Créer un objet graphique orienté :

  8. graph = nx.DiGraph() : crée un objet de graphique orienté vide.

  9. Ajouter des nœuds et des arêtes :

  10. graph.add_nodes_from(nodes) : ajoute tous les nœuds de la liste nodes au graphique.

  11. graph.add_edges_from(edges) : ajoute toutes les arêtes de la liste edges au graphique.

  12. Dessiner le graphique :

  13. nx.draw(graph, ...) : dessine le graphe à l'aide de matplotlib.

  14. with_labels=True : affiche les libellés de nœud.

  15. node_color='lightblue' : définit la couleur du nœud sur bleu clair.

  16. node_size=500 : définit la taille du nœud.

  17. font_size=10 : définit la taille de la police pour les libellés.

  18. arrowsize=20 : définit la taille des flèches pour les arêtes.

  19. Afficher le graphique :

  20. plt.show() : affiche le graphique représenté.

Résultat :

Ce code génère une visualisation de graphique orienté avec les nœuds et les arêtes spécifiés. Les nœuds sont étiquetés, de couleur bleu clair, et reliés par des flèches représentant la direction des bords.

Modèle : gemini-1.5-flash-preview-0514
Température : 1
Nombre maximal de jetons de sortie : 8 192
TopK : 40
TopP : 0,95