Автор оригинала: Pankaj Kumar.
Функция Python Numpy.square () Возвращает новый массив с элементом элемента как квадрат элементов исходного массива. Массив источника остается неизменным.
Python numpy.square () примеры
Это утилита для быстрого получения квадрата матричных элементов. Давайте посмотрим на примеры функции Numpy Square () с целочисленными, поплавками и элементами массива сложных типов.
1. Numpy Square () INT Array
import numpy as np
# ints
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
print(f'Source Array:\n{array_2d}')
array_2d_square = np.square(array_2d)
print(f'Squared Array:\n{array_2d_square}')
Выход:
Source Array: [[1 2 3] [4 5 6]] Squared Array: [[ 1 4 9] [16 25 36]]
2. Numpy Square () массив с плавающей запятой
import numpy as np
array_2d_float = np.array([1.2, 2.3, 5])
print(f'Source Array:\n{array_2d_float}')
array_2d_float_square = np.square(array_2d_float)
print(f'Squared Array:\n{array_2d_float_square}')
Выход:
Source Array: [1.2 2.3 5. ] Squared Array: [ 1.44 5.29 25. ]
Обратите внимание, что целое число в массиве с плавающей точкой преобразована в номер с плавающей запятой.
3. Numpy Square () комплексных чисел
arr = np.array([1 + 2j, 2 + 3j, 4])
print(f'Source Array:\n{arr}')
arr_square = np.square(arr)
print(f'Squared Array:\n{arr_square}')
Выход:
Source Array: [1.+2.j 2.+3.j 4.+0.j] Squared Array: [-3. +4.j -5.+12.j 16. +0.j]
Здесь целочисленный элемент преобразуется в комплексное число.
Ссылка: API док