🎨 Как нарисовать график в Python: пошаговая инструкция для начинающих

Как нарисовать граф в Python

Для того чтобы нарисовать граф в Python, вы можете использовать библиотеку Matplotlib.

import matplotlib.pyplot as plt

# Создание списков x и y
x = [1, 2, 3, 4, 5]
y = [1, 4, 9, 16, 25]

# Построение графика
plt.plot(x, y)

# Отображение графика
plt.show()

В этом примере мы создаем два списка - список x и список y, содержащих координаты точек графика. Затем мы используем функцию plot из библиотеки Matplotlib для построения графика, а функцию show для его отображения.

Как нарисовать граф в Python

Для визуализации графов в Python мы можем использовать различные библиотеки, такие как NetworkX и Matplotlib. В этой статье мы покажем вам примеры кода, которые помогут вам нарисовать графы в Python.

Установка библиотек

Прежде чем начать, необходимо установить библиотеки NetworkX и Matplotlib. Вы можете установить их, запустив следующую команду:

pip install networkx matplotlib

Пример 1: Рисование графа без атрибутов

Давайте начнем с простого примера, где мы нарисуем граф без атрибутов.

import networkx as nx
import matplotlib.pyplot as plt

# Создание пустого графа
G = nx.Graph()

# Добавление ребер в граф
G.add_edge('A', 'B')
G.add_edge('B', 'C')
G.add_edge('C', 'D')
G.add_edge('D', 'A')

# Нарисовать граф
nx.draw(G, with_labels=True)
plt.show()

Выполнив код выше, вы получите рисунок, на котором показаны вершины графа 'A', 'B', 'C' и 'D', а также ребра, соединяющие эти вершины.

Пример 2: Рисование графа с атрибутами

Теперь давайте рассмотрим пример, в котором мы предоставим атрибуты для вершин графа.

import networkx as nx
import matplotlib.pyplot as plt

# Создание пустого графа
G = nx.Graph()

# Добавление вершин с атрибутами
G.add_node('A', color='red')
G.add_node('B', color='blue')

# Добавление ребер в граф
G.add_edge('A', 'B')

# Нарисовать граф с атрибутами
colors = [G.nodes[node]['color'] for node in G.nodes]
nx.draw(G, with_labels=True, node_color=colors)
plt.show()

Выполнив код выше, вы получите рисунок, на котором вершина 'A' будет отображаться красным цветом, а вершина 'B' - синим цветом.

Пример 3: Рисование графа с различными типами вершин

В этом примере мы рассмотрим граф, в котором у вершин будут разные типы.

import networkx as nx
import matplotlib.pyplot as plt

# Создание пустого графа
G = nx.Graph()

# Добавление вершин с разными типами
G.add_node('A', node_type='circle')
G.add_node('B', node_type='square')

# Добавление ребер в граф
G.add_edge('A', 'B')

# Нарисовать граф с разными типами вершин
node_types = {node: G.nodes[node]['node_type'] for node in G.nodes}
nx.draw(G, with_labels=True, node_shape='s', node_color='lightblue', node_size=1000, nodelist=node_types.keys(), node_shape=list(node_types.values()))
plt.show()

Выполнив код выше, вы получите рисунок, на котором вершина 'A' будет отображаться в виде круга, а вершина 'B' - в виде квадрата, используя указанные типы вершин.

Заключение

Благодаря библиотекам NetworkX и Matplotlib вы легко можете создавать и визуализировать графы в Python. Начните с простых примеров, добавляйте атрибуты к вершинам и экспериментируйте с различными типами вершин, чтобы создавать более сложные и интересные визуализации графов.