Python Numpy.square() - Quadrado

1. Sintaxe do método numpy.square()
2. Códigos de exemplo: numpy.square()
3. Códigos de exemplo: numpy.square() Com out Parâmetro
4. Códigos de exemplo: numpy.square() Com Números Negativos
5. Códigos de Exemplo: numpy.square() Com Números Complexos

A função Numpy.square() calcula o quadrado de cada elemento do array dada.

É a operação inversa do método Numpy.sqrt().

Sintaxe do método numpy.square()

numpy.square(arr, out=None)

Parâmetros

Retornar

Ele retorna um array do quadrado de cada elemento do array de entrada, mesmo que seja dado out.

Códigos de exemplo: numpy.square()

import numpy as np

arr = [1, 3, 5, 7]

arr_sq = np.square(arr)

print(arr_sq)

Resultado:

[ 1  9 25 49]

Códigos de exemplo: numpy.square() Com out Parâmetro

import numpy as np

arr = [1, 3, 5, 7]
out_arr = np.zeros(4)

arr_sq = np.square(arr, out_arr)

print(out_arr)
print(arr_sq)

Resultado:

[ 1  9 25 49]
[ 1  9 25 49]

O out_arr tem a mesma forma do arr, e o quadrado do arr é salvo nele.

E o método numpy.square() também retorna a matriz quadrada, como mostrado acima.

Se o out não tem a mesma forma de arr, ele eleva um ValueError.

import numpy as np

arr = [1, 3, 5, 7]
out_arr = np.zeros(3)

arr_sq = np.square(arr, out_arr)

print(out_arr)
print(arr_sq)

Resultado:

Traceback (most recent call last):
  File "C:\Test\test.py", line 6, in <module>
    arr_sq = np.square(arr, out_arr)
ValueError: operands could not be broadcast together with shapes (4,) (3,) 

Códigos de exemplo: numpy.square() Com Números Negativos

import numpy as np

arr = [-1, -3, -5, -7]

arr_sq = np.square(arr)

print(arr_sq)

Resultado:

[ 1  9 25 49]

Códigos de Exemplo: numpy.square() Com Números Complexos

import numpy as np

arr = [1 + 2j, -2 - 1j, 2 - 3j, -3 + 4j]

arr_sq = np.square(arr)

print(arr_sq)

Resultado:

[-3. +4.j  3. +4.j -5.-12.j -7.-24.j]