The square function in NumPy allows you to perform an element-wise square of an input array. Once you provide an array, the function will return an array of similar shape with each element in the source array squared.

This function does not perform the operation in-place. Hence, the input array remains unchanged.

Function Syntax

The function syntax is as shown in the following:

 
The required parameter is x which represents the input array whose elements are what you wish to square.

Example 1: Square Int Array

In the following example, we use the square() function to square the values of an int array:

 
The given example returns an array with similar shape with each element of the input array squared.

 

Example 2: Square Floating-Point Array

You can also perform the square operation on a floating-point array as shown in the following example:

 
The resulting array is as follows:

 

Example 3: Working with Complex Numbers

The square function also allows you to perform the square operations on complex numbers as shown in the following example:

 
The function returns the square of the provided array as complex numbers.

 

Conclusion

In this brief article, we covered how to use the NumPy square function to get the square of each element in the input array.