numpy.square

numpy.square(x, *ufunc_args) = <ufunc 'square'>

Функция square() вычисляет квадрат элементов массива, т.е. каждый элемент массива умножается сам на себя.

Данная функция эквивалентна команде x**2.

Однако, numpy.square() является универсальной функцией, т.е. поддерживает целый ряд параметров, которые позволяют оптимизировать ее работу в зависимости от специфики алгоритма, в котором она необходима.

Параметры:
x - число, массив или подобный массиву объект
Входные данные.
*ufunc_args - аргументы универсальной функции
Аргументы, позволяющие настроить и оптимизировать работу функции (подробнее см. универсальные функции).
Возвращает:
результат - массив NumPy или вещественное число
Массив вычисленных квадратов от значений из x или число, если на вход подано одно число.
Смотрите так же: sqrt, power, float_power

Примеры

>>> import numpy as np
>>> 
>>> np.square(7)
49
>>> 
>>> np.square([1, 2, 3, 4, 5])
array([ 1,  4,  9, 16, 25], dtype=int32)
>>> 
>>> a = np.arange(9).reshape(3, 3)
>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> 
>>> np.square(a)
array([[ 0,  1,  4],
       [ 9, 16, 25],
       [36, 49, 64]], dtype=int32)
>>> 
>>> np.square(2 + 2j)
8j
>>>
>>> np.square([2 - 1j, 3 - 2j, 4 - 3j])
array([3. -4.j, 5.-12.j, 7.-24.j])