In NumPy, the square() function is used to compute the element-wise square of each element in an input array. It returns a new array with the squares of the original elements. The function is part of the NumPy mathematical functions and is particularly useful for operations where you need to square each element in an array.
In other words, this mathematical function helps to user calculate the square values of every element in the given array. The NumPy square()
function takes four parameters arr
, out
, where
, and dtype
, and returns a new array with an argument value as the square of the source array elements. In this article, I will explain how to use Numpy square()
function on single and multi-dimension arrays.
1. Quick Examples of NumPy square() Function
If you are in a hurry, below are some quick examples of how to use Python NumPy square() function.
# Quick examples of numpy square() function
# Example 1: Use numpy.square() function
# To get single element
arr = 25
arr2 = np.square(arr)
# Example 2: Use numpy.square() function
arr = [2, 3, 5, 8, 9, 4]
arr2 = np.square(arr)
# Example 3: Use numpy.square() function
# To floating-point array
arr = [2.3, 3.5, 5.1, 8.7, 9.2, 4.8]
arr2 = np.square(arr)
# Example 4: Use numpy.square() function
# With negative numbers
arr = [-6, -13, -25, -8, -15, -17]
arr2 = np.square(arr)
# Example 5: Use numpy.square() function
# To square from multiple array
arr = np.array([[5, 8, 3, 7], [9, 4, 2, 6]])
arr2 = np.square(arr)
# Example 6: Use numpy.square() function
# with complex numbers
arr = [2+3j, -4-5j, 3-4j, -3+4j]
arr2 = np.square(arr)
# Example 7: Use complex numbers
arr = 6 + 5j
arr2 = np.square(arr)
# Example 8: Use numpy.square() function
# To graphical representation
arr = np.linspace(start = -12, stop = 15,num = 6, endpoint = True)
arr2 = np.square(arr)
plt.plot(arr, np.square(arr))
plt.plot(arr, arr, color = 'yellow')
plt.show()
# Example 9: Use numpy.square()
# To get the empty array
arr = np.array([])
arr2 = np.square(arr)
2. Syntax of NumPy square()
Following is the syntax of the numpy.square()
function.
# Syntax of numpy.square()
numpy.square(arr, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None)
2.1 Parameters of square()
arr
– Input array.out
– It is ndarray, None, or tuple of ndarray and None, optional. Out will be the location where the result is to be stored. If provided, it must have a shape that the inputs broadcast to. If not provided or None, a freshly allocated array is returned.where
(optional) – A boolean array of the same shape asarr
, where an element is squared only if the corresponding value isTrue
. This condition is broadcast over the input.dtype
(optional) – The desired data type for the array. If not specified, the data type ofarr
is used.
2.2 Return Value
It returns an array with square() value of each input array
3. Usage of NumPy square() Function
This mathematical Python NumPy in-built function square()
takes the input array as param and returns a square value of each element in the input array. Moreover, you can get the square root values of an array using numpy.sqrt()
.
3.1 Get the Square Value of Single Element
You can use the numpy.square()
function to compute the square value of a single element of an input NumPy array. For instance, first import the NumPy library with the alias np
. Creates a NumPy array arr
containing a single element, 25
. Then you can use np.square()
to compute the square of the single element in the array.
# Import numpy
import numpy as np
# Create input array
arr = np.array([25])
print("Original array:",arr)
# Use numpy.square()
# To get the square value
arr2 = np.square(arr)
print("Square value:",arr2)
Yields below output.
3.2 Get the Square Values of Multiple Elements of 1-D Array
If you want to get the square values of multiple elements in a 1-D array using NumPy. You can calculate the square values of integer array elements by using the numpy.square()
function.
In the below example, to create a 1-D NumPy array arr
with elements [2, 3, 5, 8, 9, 4]
, then use np.square()
to compute the square of each element in the array. Finally, it prints both the original array and the resulting array with squared values.
# Create an 1D input array
arr = [2, 3, 5, 8, 9, 4]
print("Original array:",arr)
# Use numpy.square() function
# To get the square value
arr2 = np.square(arr)
print("Square values:",arr2)
Yields below output.
4. Get the Square Values of Float Elements of 1-D NumPy Array
If you have a 1-D NumPy array with floating-point elements and you want to get the square values of those elements. For instance, arr
is a 1-D NumPy array with float elements [2.3, 3.5, 5.1, 8.7, 9.2, 4.8]
. The np.square()
function is then used to compute the square of each element in the array.
# Creating an 1D input array
arr = [2.3, 3.5, 5.1, 8.7, 9.2, 4.8]
print("Original array:",arr)
# Use numpy.square() function
# To get the square values of floating point
arr2 = np.square(arr)
print("Square values:",arr2)
# Output:
# Original array: [2.3, 3.5, 5.1, 8.7, 9.2, 4.8]
# Square values: [ 5.29 12.25 26.01 75.69 84.64 23.04]
5. Get the Square Values of Negative Elements
You can also calculate the square values of the negative numbers using the numpy.square()
function. To get the square values of negative elements in a NumPy array, For example, arr
is a 1-D NumPy array with negative elements [-6, -13, -25, -8, -15, -17]
. The np.square()
function is used to compute the square of each element in the array.
# Creating an 1D input array
arr = [-6, -13, -25, -8, -15, -17]
print("Original array:",arr)
# Use numpy.square() function
# With negative numbers
arr2 = np.square(arr)
print("Square values:",arr2)
# Output:
# Original array: [-6, -13, -25, -8, -15, -17]
# Square values: [ 36 169 625 64 225 289]
6. Get the Square Values of 2-D NumPy Array Elements
If you want to get square values of 2-dimensional array elements you will use square()
function of NumPy array. It will give the square values of 2-D array elements. Let’s create a 2-D NumPy array using numpy.array()
.
# Creating an 2D input array
arr = np.array([[5, 8, 3, 7], [9, 4, 2, 6]])
# Use numpy.square() function
# Get the square values of 2-D array elements
arr2 = np.square(arr)
print("Square values of 2D array:\n",arr2)
# Output:
# Square values of 2D array:
# [[25 64 9 49]
# [81 16 4 36]]
7. Get the Square Values of Complex Numbers
If you have a NumPy array containing complex numbers and you want to get the square values of those complex numbers, you can use the numpy.square()
function.
In the below example, arr
is a 1-D NumPy array with complex numbers. The np.square()
function is then used to compute the square of each complex number in the array.
# Create an array with complex values
arr = [2+3j, -4-5j, 3-4j, -3+4j]
# Use numpy.square() function
# With complex numbers
arr2 = np.square(arr)
print("Square values:\n",arr2)
# Output :
# Square values:
# [-5.+12.j -9.+40.j -7.-24.j -7.-24.j]
8. Get the Square Values on Graphical Representation
You can also use NumPy square()
function for graphical representation using MatLab library.
import numpy as np
import matplotlib.pyplot as plt
# Use numpy.square() function to graphical representation
arr = np.linspace(start = -12, stop = 15,
num = 6, endpoint = True)
arr2 = np.square(arr)
print(arr2)
plt.plot(arr, np.square(arr))
plt.plot(arr, arr, color = 'yellow')
plt.show()
# Output:
# [144. 43.56 1.44 17.64 92.16 225. ]
You can see the Parabolic graph of the square() method in Numpy.
9. Get The Empty Array
To create an empty NumPy array using np.array([])
and then apply the np.square()
function to it. However, it’s important to note that an empty array, in this context, refers to an array with zero elements.
In this case, np.square()
is applied to each element of the empty array, resulting in another empty array. The squaring operation doesn’t change the fact that the array is empty.
# Create an empty array
arr = np.array([])
# Use numpy.square()
# To get the empty array
arr2 = np.square(arr)
print("Square empty array:\n",arr2)
# Output:
# Square empty array:
# []
Frequently Asked Questions
The numpy.square()
function is a NumPy mathematical function that is used to compute the element-wise square of each element in a given array. It takes an array as input and returns a new array where each element is the square of the corresponding element in the input array.
numpy.square()
is designed to work with arrays of different data types, including integers, floats, and complex numbers.
numpy.square()
can handle complex numbers. When applied to an array containing complex numbers, it computes the square of each complex number, which involves squaring both the real and imaginary parts.
If the input array passed to numpy.square()
is empty, the function will return an empty array as well. The resulting array will have the same shape as the input array, which, in the case of an empty array, means it will also be empty.
numpy.square()
handles negative values by squaring each element independently, without considering the sign. When you square a negative number, the result is positive.
You can use numpy.square()
with arrays resulting from mathematical expressions. The function operates element-wise on the input array, so any array resulting from a mathematical expression can be squared using numpy.square()
Conclusion
In this article, I have explained how to use Python NumPy.square()
function using how to calculate the square value of every element in the given array with examples.
Happy Learning!!
Related Articles
- How to convert NumPy array to list
- How to Slice NumPy Array?
- Python NumPy Array Reshape
- Get NumPy Array Length
- Python NumPy Array Copy
- How to Use NumPy argmax in Python
- How to create NumPy array in different ways ?
- Python NumPy zeros() Function
- How to use Python NumPy arange() Function
- Python NumPy repeat() Function