Python Visualization

Write Python code to visualize a weighted directed graph with example data

You can query a model directly and test the results returned when using different parameter values with the Cloud console, or by calling the Vertex AI API directly. For an example of using the Vertex AI API, see Quickstart using the Vertex AI API. To view this sample in the Cloud console: Go to Google Cloud console

Prompt Freeform Write Python code to draw a directed graph using matplotlib and networkx. Show how to use the code with some example data. Include all necessary import statements. Response 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() Explanation: Import necessary libraries: matplotlib.pyplot for plotting the graph. networkx for creating and manipulating the graph. Define example data: nodes: A list of node names. edges: A list of tuples representing directed edges, where each tuple is (source node, target node). Create a directed graph object: graph = nx.DiGraph(): Creates an empty directed graph object. Add nodes and edges: graph.add_nodes_from(nodes): Adds all nodes from the nodes list to the graph. graph.add_edges_from(edges): Adds all edges from the edges list to the graph. Draw the graph: nx.draw(graph, ...): Draws the graph using matplotlib. with_labels=True: Displays node labels. node_color='lightblue': Sets the node color to light blue. node_size=500: Sets the node size. font_size=10: Sets the font size for labels. arrowsize=20: Sets the arrow size for edges. Display the graph: plt.show(): Shows the plotted graph. Output: This code will generate a directed graph visualization with the specified nodes and edges. The nodes will be labeled, colored light blue, and connected by arrows representing the direction of the edges.

Model: gemini-1.5-flash-001 Temperature: 1 Max output tokens: 8192 TopK: 40 TopP: 0.95