Векторизация функции Python в NumPy

Обёртка vectorize принимает функцию Python и возвращает объект, который можно вызывать на массивах: он пройдёт по ячейкам и соберёт ответ. Это удобство, а не быстрая встроенная универсальная функция.

Зададим функцию, которая удваивает число, и применим её к ряду оценок:

import numpy as np

def double(x):
	return x * 2

twice = np.vectorize(double)
marks = np.array([1, 2, 3])
print(twice(marks)) # выведет [2 4 6]

Напишите функцию, которая прибавляет 5 к числу, оберните её и примените к ряду цен [10, 20], затем выведите результат.

Сделайте функцию «квадрат оценки», примените её к [2, 3, 4] через обёртку и выведите ряд.