matplotlib. написал код но он не работает, причём при вписании кода с интернета всё хорошо

import matplotlib.pyplot as plt #Імпортуємо matplotlib, модуль pyplot під псевдонімом plt
print("Практичне завдання №10") #Номер практичного заняття

prizv = "Шутенко"               #Вказуємо прізвище
imya = "Андрій"             #Вказуємо ім'я
grupa = "ОТ-11"                #Вказуємо групу
variant = "8"                  #Вказуємо варіант
#Створюємо список
inici = [prizv, imya, grupa, variant]
#Виводимо завдання
print("Визначити площу бічної, повної поверхні, та об’єм циліндра.")
#Вводимо параметри завдання(Радіуси основ циліндра)
r = float(input("Введіть радіус  основи циліндра, r = "))
h = float(input("Введіть висоту циліндра, h = "))
#Вводимо додаткові умови, так як змінні не можуть дорівнювати 0 або від'ємному числу за умовою задачі
if (r*h)<=0:
    print("Помилка")

else:
    # Вводимо число Пі
    pi = 3.14
    # Формула розрахунку площі бічної поверхні
    s1 = 2 * pi * r * h
    # Формула розрахунку площі повної поверхні
    s2 = 2 * pi * r * (r + h)
    # Формула знаходження об’єму
    v = pi * r ** 2 * h
    # Виводимо результати
    print("площа бічної поверхні =", s1, "площа повної поверхні =", s2, "об’єм =", v)
    q = float(input())

# створюємо список roz(містить дані для діаграми)
roz = [s2, v]
# Задаємо х значення b_1
x = s2
# Задаємо y значення d
y = v
# Змінна fig представляє весь рисунок.Змінна ах репрезентує одну діаграму на рисунку
fig, ax = plt.subplots()
# викликаємо метод plot який побудує діаграму на основі наданих даних
ax.plot(roz)

# Функція plt.show відкриває переглядач Matplotlib і відображає візуалізацію
plt.show()