Python Numpy.square() - Square

1. Syntax of numpy.square()
2. Example Codes: numpy.square()
3. Example Codes: numpy.square() With the out Parameter
4. Example Codes: numpy.square() With Negative Numbers
5. Example Codes: numpy.square() With Complex Numbers

Numpy.square() function calculates the square of every element in the given array.

It is the inverse operation of Numpy.sqrt() method.

Syntax of numpy.square()

numpy.square(arr, out=None)

Parameters

Return

It returns an array of the square of each element in the input array, even if out is given.

Example Codes: numpy.square()

import numpy as np

arr = [1, 3, 5, 7]

arr_sq = np.square(arr)

print(arr_sq)

Output:

[ 1  9 25 49]

Example Codes: numpy.square() With the out Parameter

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)

Output:

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

out_arr has the same shape as arr, and the square of arr is saved in it.

And the numpy.square() method also returns the square array, as shown above.

If out doesn’t have the same shape as arr, it raises a 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)

Output:

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,) 

Example Codes: numpy.square() With Negative Numbers

import numpy as np

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

arr_sq = np.square(arr)

print(arr_sq)

Output:

[ 1  9 25 49]

Example Codes: numpy.square() With Complex Numbers

import numpy as np

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

arr_sq = np.square(arr)

print(arr_sq)

Output:

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