API · NumPy

Python Numpy.square() - Cuadrado

La función Python numpy.square() calcula el cuadrado de un array numpy.

En esta página

La función Numpy.square() calcula el cuadrado de cada elemento de el array dada.

Es la operación inversa del método Numpy.sqrt().

La sintaxis de numpy.square()

numpy.square(arr, out=None)

Parámetros

arr matriz de entrada
out Si se da out, el resultado se almacenará en out. out debería tener la misma forma que arr.

Retorna

Devuelve un array del cuadrado de cada elemento de el array de entrada, incluso si se da out.

Códigos de ejemplo: 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 ejemplo: numpy.square() Con 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]

out_arr tiene la misma forma que arr, y el cuadrado de arr está guardado en él.

Y el método numpy.square() también devuelve el array cuadrada, como se muestra arriba.

Si out no tiene la misma forma que arr, genera un 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 ejemplo: numpy.square() Con 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 ejemplo: numpy.square() Con números complejos

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]