`numpy.square()`

Функция np.square поэлементно вычисляет квадрат каждого элемента, то есть x*x. Эквивалент np.power(x, 2) и оператора x**2, но обычно немного быстрее.

Сигнатура

numpy.square(x, /, out=None, *, where=True, dtype=None)

Параметры

x — входной массив или число.
out — необязательный выходной массив.
where — условная маска.

Возвращаемое значение

Массив той же формы, что и x, со значениями x**2.

Примеры

Пример 1. Квадраты целых чисел.

import numpy as np

print(np.square([1, 2, 3, 4, 5]))

[ 1  4  9 16 25]

Пример 2. Энергия дискретного сигнала.

signal = np.array([0.5, -0.3, 0.8, -0.6, 0.2])
energy = np.sum(np.square(signal))
print(round(energy, 3))

1.38

Пример 3. Преобразование расстояния в квадрат расстояния (быстрее, чем sqrt).

dx = np.array([3, 6, 1])
dy = np.array([4, 8, 1])
dist_sq = np.square(dx) + np.square(dy)
print(dist_sq)

[ 25 100   2]

Пример 4. Квадрат для отрицательных значений.

print(np.square([-2, -1, 0, 1, 2]))

[4 1 0 1 4]

Пример 5. Дисперсия через np.square.

data = np.array([2.0, 4.0, 4.0, 4.0, 5.0, 5.0, 7.0, 9.0])
var = np.mean(np.square(data - np.mean(data)))
print(var)

4.0

См. также

Примечание

Лицензия и источники

Техническое описание функции адаптировано из официальной документации NumPy (https://numpy.org/doc/stable/), BSD-3-Clause License. Примеры и пояснения — © AlashEd Wiki.

numpy.square()