Visualização em Python

Escrever um código Python para visualizar um gráfico direcionado ponderado com dados de exemplo

É possível consultar um modelo diretamente e testar os resultados retornados ao usar diferentes valores de parâmetros com o console do Cloud ou ao chamar a API Vertex AI diretamente.

Comando Formato livre

Escreva um código Python para desenhar um gráfico direcionado usando matplotlib e networkx. Mostre como usar o código com alguns dados de exemplo. Inclua todas as instruções de importação necessárias.

Resposta


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()

Explicação:

  1. Importe as bibliotecas necessárias:
  2. matplotlib.pyplot para representar o gráfico.
  3. networkx para criar e manipular o gráfico.

  4. Defina dados de exemplo:

  5. nodes: uma lista de nomes de nós.

  6. edges: uma lista de tuplas que representam arestas dirigidas, em que cada tupla é (nó de origem, nó de destino).

  7. Crie um objeto de gráfico direcionado:

  8. graph = nx.DiGraph(): cria um objeto de gráfico dirigido vazio.

  9. Adicionar nós e bordas:

  10. graph.add_nodes_from(nodes): adiciona todos os nós da lista nodes ao gráfico.

  11. graph.add_edges_from(edges): adiciona todas as arestas da lista edges ao gráfico.

  12. Desenhe o gráfico:

  13. nx.draw(graph, ...): desenha o gráfico usando o matplotlib.

  14. with_labels=True: mostra os rótulos dos nós.

  15. node_color='lightblue': define a cor do nó como azul-claro.

  16. node_size=500: define o tamanho do nó.

  17. font_size=10: define o tamanho da fonte dos rótulos.

  18. arrowsize=20: define o tamanho da seta para as bordas.

  19. Mostrar o gráfico:

  20. plt.show(): mostra o gráfico traçado.

Saída:

Esse código vai gerar uma visualização de gráfico dirigido com os nós e as arestas especificados. Os nós serão rotulados, coloridos em azul claro e conectados por setas que representam a direção das arestas.

Modelo: gemini-1.5-flash-001
Temperatura: 1
Máximo de tokens de saída: 8192
TopK: 40
TopP: 0,95