numpy.squeeze() in NumPy

Parameters for numpy.squeeze() method

The parameters that are involved in the numpy.squeeze method are:

• Array: This is the array of numbers we will work with. It is compulsory option.
   
• Axis: This selects a subset of the length in the given shape. It is an optional parameter.

Return Value for numpy.squeeze() method

The squeeze() method returns the squeezed ndarray array. Squeezed [ndarray] means removing extra single dimensions from the input array. It's either the array itself or a way to look at the same data more compactly.

Example of numpy.squeeze() method

Here are some examples to understand the working of the numpy.squeeze method.

Compress a One-Dimensional Element within an Array using numpy.squeeze()

In this example, we will use the numpy.squeeze() method to compress a one-dimensional element within an Array.

Code

• Python

Python

import numpy as np

# Create a one-dimensional element
cn_score = np.array([[[4, 2, 7]]])

# Squeeze the array to remove dimensions of size 1
compressed_array = np.squeeze(cn_score)

print("Original Array:")
print(cn_score)
print("Shape of Original Array:", cn_score.shape)

# Print the Compressed array
print("\nCompressed Array:")
print(compressed_array)
print("Shape of Compressed Array:", compressed_array.shape)

You can also try this code with Online Python Compiler

Run Code

Output

Explanation

This code operates NumPy for array manipulations. It creates a 3D array 'cn_score' with elements [4, 2, 7]. Using squeeze() shrinks it to 1D. Displaying the original array and compressed array, we observe that the shape changes from (1, 1, 3) to (3) after using the numpy.squeeze() method. NumPy reduces array handling in Python, increasing efficiency.

Compress multi-dimensional Elements within an Array using numpy.squeeze() method.

In this example, we will use the squeeze() method to compress a multi-dimensional element within an Array.

Code

• Python

Python

import numpy as np

# Create an array with multi-dimensional elements
cn_array = np.array([[[[5, 1, 7]]], [[[2, 8, 9]]]])

# Squeeze the array to remove dimensions of size 1
comp_array = np.squeeze(cn_array)

print("Original Array:")
print(cn_array)
print("Original Array Shape:", cn_array.shape)
print("\nCompressed Array:")
print(comp_array)

You can also try this code with Online Python Compiler

Run Code

Output

Explanation

This code operates the NumPy library. It creates a multi-dimensional array called 'cn_array' and then squeezes it using np.squeeze(). It prints the output and shape of the original_array and the result of the compressed_array. The original 'cn_array' has a shape (2, 1, 1, 3), while the squeezed 'comp_array' has a shape of (2, 3), displaying the difference.

Use numpy.squeeze along a specific Axis.

We will use the numpy.squeeze() method along a specific axis in this example.

Code

• Python

Python

import numpy as np

# Create an array
cn_array = np. array ([[[[7], [2], [1]]],
                                   [[[3], [8], [5]]]])

# Squeezing along axis 1
comp_array_axis1 = np.squeeze(cn_array, axis=1)

# Squeezing along axis 3
comp_array_axis3 = np.squeeze(cn_array, axis=3)

print("Original Array:")
print(cn_array)
print("Original Array Shape:", cn_array.shape)
print("\nCompressed Array (Axis 1):")
print(comp_array_axis1)
print("Compressed Array Shape (Axis 1):", comp_array_axis1.shape)
print("\nCompressed Array (Axis 3):")
print(comp_array_axis3)
print("Compressed Array Shape (Axis 3):", comp_array_axis3.shape)

You can also try this code with Online Python Compiler

Run Code

Output

Explanation

The NumPy library is used in this Python code to manipulate arrays. Initially, it constructs a multi-dimensional array named 'cn_array.' Then, it uses the 'np.squeeze' function to remove dimensions with a size of one along specified axes. This process generates 'comp_array_axis1' by squeezing along axis one and 'comp_array_axis3' along axis 3, eliminating the respective size-1 dimensions.

Frequently Asked Questions

What does the `numpy.squeeze()` characteristic do?

The numpy.squeeze() function in NumPy removes single-dimensional entries (axes) from the shape of an array. It reduces the array's dimensionality by disposing of dimensions with size 1.

How is the `numpy.squeeze()` feature used?

The function is usually called on a NumPy array and may take arguments: the 'array' and the `axis` parameter. The `axis` parameter specifies the size to cast off the dimensions.

What's the cause of using `numpy.squeeze()`?

Using `numpy.squeeze()` mainly aims to simplify array shapes and decrease pointless dimensions. It is especially beneficial while operating with statistics with greater dimensions that don't contribute to the actual means or computation.

Are there any risks or troubles while using `numpy.squeeze()`?

One needs to be careful when using `numpy. squeeze()` because removing dimensions might result in unintentional records manipulation.

Can `numpy.squeeze()` result in facts loss?

Using `numpy.squeeze()` can potentially cause records loss without the right array shape expertise. Examining the information before and after using the characteristic is essential to ensure no crucial information is discarded.

Conclusion

In this article, we learned about the Squeeze function in NumPy. We explore different aspects related to squeezing, like its syntax, parameters, and return value for numpy.squeeze(). We learned that two parameters are involved in the squeeze method: "array" and "axis." We also learned about squeeze() working with the help of different examples like one-dimensional arrays, multi-dimensional arrays, and many more.

Check out the link to learn more about such a topic.

• NumPy.Reshape() in Python
• NumPy.Transpose() in Python
• NumPy Basics
• Python packages
   

You can find more informative articles or blogs on our platform. You can also practice more coding problems and prepare for interview questions from well-known companies on your platform, CodingNinjasStudio.