Функция square

Функция square из модуля numpy возвращает новый массив, в котором каждый элемент исходного массива возведён в квадрат. Функция принимает один обязательный параметр - массив чисел, и один необязательный - тип данных для результата.

Синтаксис

np.square(arr, [dtype])

Пример ⁅n="sect"⁆

Давайте возведём в квадрат каждый элемент одномерного массива:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
res = np.square(arr)
print(res)

Результат выполнения кода:

[ 1 4 9 16 25]

Пример ⁅n="sect"⁆

Функция также работает с отрицательными числами:

import numpy as np

arr = np.array([-3, -2, -1, 0, 1, 2, 3])
res = np.square(arr)
print(res)

Результат выполнения кода:

[9 4 1 0 1 4 9]

Пример ⁅n="sect"⁆

Возведение в квадрат элементов двумерного массива:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.square(arr)
print(res)

Результат выполнения кода:

[[ 1  4  9]
 [16 25 36]]

Пример ⁅n="sect"⁆

Укажем тип данных для результата, например, float64:

import numpy as np

arr = np.array([1, 2, 3])
res = np.square(arr, dtype=np.float64)
print(res)

Результат выполнения кода:

[1. 4. 9.]

Пример ⁅n="sect"⁆

Используем функцию с массивом, содержащим числа с плавающей точкой:

import numpy as np

arr = np.array([1.5, 2.5, 3.5])
res = np.square(arr)
print(res)

Результат выполнения кода:

[ 2.25 6.25 12.25]

Смотрите также

функцию sqrt,
которая вычисляет квадратный корень элементов массива
функцию power,
которая возводит элементы массива в произвольную степень
функцию reciprocal,
которая вычисляет обратные значения элементов массива
функцию absolute,
которая вычисляет абсолютные значения элементов массива