How does numpy.square() work?

NumPy is a high-functioning Python library that provides a lot of tools for array manipulation. It makes our lives as developers a whole lot easier.

The NumPy square() function is a function in the NumPy package that helps us to calculate the square value of every element in a NumPy array/list. Apart from calculating squares of array elements, we can calculate squares of other data structures too.

In this article, we will go over the basics of NumPy square(), as well as some examples in which we use NumPy square() to generate squares of array elements.

Syntax for NumPy square()

The syntax for NumPy square() function is:

numpy.square(arr, out=None, where=True, dtype=None)

Parameters for NumPy square()

The parameters for NumPy square() function are:

• arr: This mandatory parameter represents the input data.
• out: This optional parameter represents the location in memory where the result will be stored.
• where: This optional parameter represents the condition that is applied to the input data.
• dtype: This optional parameter represents the data type of the input data.

Return Type for NumPy square()

The NumPy square() function returns an array with the square value of each element of the input array.

Examples

To understand the working of the NumPy square() function in a better sense, let's go through some practical examples:

Using NumPy square() Function to Get Single Element

With the help of the NumPy square() function, we can calculate the square of a single element as input. To perform this, we need to pass an array with a single element as the number, of which we need to find the square.

Output

Using NumPy square() Function to Floating-Point Array

Just for an array, we can find the squares of every element in a floating-point array.

Output

Using NumPy square() Function with Negative Numbers

We can find the square value of all the negative elements in an array using the NumPy square() function.

Output

Using NumPy square() to Get Squares of Multiple Elements of an Array

With the help of the NumPy square() function, we are able to calculate the squares of every element of an array.

Output

Using NumPy square() Function to get Square Value of an Empty Array

When we use the NumPy square() function on an empty array, an empty array is returned.

Output

Using NumPy square() Function to Get Squares of Complex Numbers

We can use the NumPy square() function to calculate square values of complex numbers. Complex numbers are numbers that are represented by a+bj format. (a and b being integers).

Output

Visualizing NumPy square() Function Graphically

Matplotlib is a very powerful graphic library in Python, which allows us to create illustrative and exemplary graphs and plots. Using this library, we can easily visualize the NumPy square() function.

Output