Рисуем график функции средствами Matplotlib на Python

Сегодня мы будем на языке программирования python строить графики по точкам, полученным из уравнения. С помощью библиотеки matplotlib.

Для начала установим необходимые библиотеки.
На Mac мне пришлось выполнить эти команды:

brew upgrade freetype
brew install pkg-config

Потом установим саму библиотеку, она сама установит сразу все зависимости.
pip3 install matplotlib.

Хочу отметить, если у вас будет ругаться на tkinter, то установите это расширение
sudo apt-get install python3-tk

Наше уровнение будет следующим:
y=x^2

По нему и будем строить график.

Для начала импортируем matplotlib

import matplotlib.pyplot as plot

Реализуем саму функцию, которая будет принимать ряд чисел, они будут координатами по x.

def math_expr(*args):

Объявим две переменные типа list (списки)

data_x = []
data_y = []

которые будут отвечать за координаты x и y соответственно.

Далее проходим циклом по массиву с координатами x, который пришел к нам в виде аргумента.

    for x in args:
        data_x.append(x)
        data_y.append(x**2)

В цикле мы заполним два массива(data_x и data_y) значениями x и x в квадрате.

Далее мы "заполняем" matplotlib нашими данными, присваиваем label и выводим на экран.

plot.plot(data_x, data_y)
plot.ylabel('детское выражение')
plot.show()

Так просто в 12 строчек кода python позволяет построить график функции.

Материал подготовлен автором @zheev

1081