Збереження малюнку

Малюнок можна зберегти у файлі

x = np.linspace(-1.4, 1.4, 30)
plt.plot(x, x**2)
plt.savefig("my_square_function.png", transparent=True) 

Стиль ліній та колір

За замовчуванням matplotlib малює лінію між послідовними точками.

plt.plot([0, 100, 100, 0, 0, 100, 50, 0, 100],
         [0, 0, 100, 100, 0, 100, 130, 100, 0])
plt.axis([-10, 110, -10, 140])
plt.show()  

Щоб змінити стиль і колір лінії потрібно перелати третій аргумент. Наприклад, «g--» означає «зелена пунктирна лінія».

plt.plot([0, 100, 100, 0, 0, 100, 50, 0, 100],
         [0, 0, 100, 100, 0, 100, 130, 100, 0],
         "g--")
plt.axis([-10, 110, -10, 140])
plt.show() 

Можна побудувати кілька ліній на одному графіку вказавши стиль кожної лінії так

plt.plot([0, 100, 100, 0, 0], [0, 0, 100, 100, 0], "r-",
         [0, 100, 50, 0, 100], [0, 100, 130, 100, 0], "g--")
plt.axis([-10, 110, -10, 140])
plt.show()  

або так

plt.plot([0, 100, 100, 0, 0], [0, 0, 100, 100, 0], "r-")
plt.plot([0, 100, 50, 0, 100], [0, 100, 130, 100, 0], "g--")
plt.axis([-10, 110, -10, 140])
plt.show() 

Можна малювати прості точки замість ліній. Приклад із зеленими рисками, червоною пунктирною лінією та синіми трикутниками.

x = np.linspace(-1.4, 1.4, 30)
plt.plot(x, x, 'g--', x, x**2, 'r:', x, x**3, 'b^')
plt.show() 

Функція plot повертає список об’єктів Line2D (по одному для кожної лінії). Для цих ліній можна встановити додаткові властивості, наприклад ширину лінії, стиль тире або рівень альфа. Перегляньте повний список властивостей у документації.

x = np.linspace(-1.4, 1.4, 30)
line1, line2, line3 = plt.plot(x, x, 'g--', x, x**2, 'r:', x, x**3, 'b^')
line1.set_linewidth(3.0)
line1.set_dash_capstyle("round")
line3.set_alpha(0.2)
plt.show() 

Стиль та колір ліній можна вказувати явно можна використати аргумент ключового слова linestyle або, коротше, ls:

x = np.linspace(-1.4, 1.4, 30)
ypoints = np.array([3, 8, 1, 10])
plt.plot(ypoints, linestyle = 'dotted')
plt.show() 

Основні стилі

Перший графік

Найпростіший графік

import matplotlib.pyplot as plt
plt.plot([1, 2, 4, 9, 5, 3])
plt.show()
  

Просто передаємо функції `plot` список з даними, а потім викликаємо функцію show! Якщо функції plot надано один масив даних, вона використовуватиме його як координати на вертикальній осі, а індекс кожної точки даних у масиві використовуватиме як горизонтальну координату. Також можете надати два масиви: один для горизонтальної осі x, а другий для вертикальної осі y:

plt.plot([-3, -2, 5, 0], [1, 6, 4, 3])
plt.show()
  

Осі автоматично підлаштовуються під діапазон даних, але діапазон осей можна встановити вручну у форматі [xmin, xmax, ymin, ymax].

plt.plot([-3, -2, 5, 0], [1, 6, 4, 3])
plt.axis([-4, 6, 0, 7])
plt.show()  

Щоб побудуємо графік явно заданої математичної функції $y=x^2$, створимо масив $x$ із 500 числами з плаваючою точкою в діапазоні від -2 до 2, а потім створюємо другий масив із квадратів елеементів масиву $x$

import numpy as np
x = np.linspace(-2, 2, 500)
y = x**2
plt.plot(x, y)
plt.show() 

Додамо назву, мітки осей і намалюємо сітку.

plt.plot(x, y)
plt.title("Графік функції")
plt.xlabel("x")
plt.ylabel("$y = x^2$")
plt.grid(True)
plt.show()