📊 Как построить диаграмму рассеивания в Python? Изучаем легкими шагами

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

import matplotlib.pyplot as plt

# Данные для оси x и y
x = [1, 2, 3, 4, 5]
y = [2, 4, 6, 8, 10]

# Построение диаграммы рассеивания
plt.scatter(x, y)

# Настройка осей и заголовка
plt.xlabel('Ось x')
plt.ylabel('Ось y')
plt.title('Диаграмма рассеивания')

# Отображение диаграммы
plt.show()

Как построить диаграмму рассеивания в Python?

Диаграмма рассеивания (scatter plot) - это график, который показывает распределение двух переменных и их взаимосвязь. Этот тип графика особенно полезен для визуализации показателей, которые измеряются на числовой шкале.

Шаг 1: Установка библиотеки Matplotlib

Перед тем, как начать построение диаграммы рассеивания, убедитесь, что у вас установлена библиотека Matplotlib. Если она еще не установлена, выполните команду:

    !pip install matplotlib
    

Шаг 2: Импорт необходимых модулей

После установки Matplotlib, импортируйте необходимые модули:

    import matplotlib.pyplot as plt
    import numpy as np
    

Шаг 3: Создание данных

Прежде чем построить диаграмму рассеивания, необходимо создать данные для него. Возьмем пример с двумя переменными X и Y:

    X = np.array([1, 2, 3, 4, 5])
    Y = np.array([6, 7, 8, 9, 10])
    

Шаг 4: Построение диаграммы рассеивания

Теперь мы готовы построить диаграмму рассеивания. Используйте функцию scatter из модуля pyplot следующим образом:

    plt.scatter(X, Y)
    plt.xlabel('X')
    plt.ylabel('Y')
    plt.title('Диаграмма рассеивания')
    plt.show()
    

В данном примере X представляет значения по оси X, а Y - значения по оси Y. Функции xlabel, ylabel и title используются для добавления подписей к осям и заголовка к графику соответственно.

Пример с реальными данными

Теперь рассмотрим пример с реальными данными. Допустим, у нас есть данные о росте и весе нескольких человек. Мы хотим посмотреть, как связаны эти две переменные. Прежде чем построить диаграмму, создадим массивы с данными:

    height = np.array([160, 165, 170, 175, 180])
    weight = np.array([60, 65, 70, 75, 80])
    

Теперь построим диаграмму рассеивания, используя эти данные:

    plt.scatter(height, weight)
    plt.xlabel('Рост (см)')
    plt.ylabel('Вес (кг)')
    plt.title('Диаграмма рассеивания: Рост vs Вес')
    plt.show()
    

Это приведенный выше код построит график рассеивания, где по оси X отображается рост, а по оси Y - вес. Полученная диаграмма поможет нам увидеть, существует ли какая-либо зависимость между этими двумя переменными.

Заключение

Построение диаграммы рассеивания в Python с помощью библиотеки Matplotlib - простой способ визуализации взаимосвязи между двумя переменными. При использовании функции scatter из модуля pyplot, вы можете легко настраивать свои диаграммы и добавлять подписи, улучшая их визуальное представление.