If one or more of your axes are of length one, you can remove those axes using numpy.squeeze.
np.squeeze: Remove Axes of Length One From an Array
Stay up-to-date with
data skills using
CodeCut
CodeCut is a platform that offers short and visually appealing code snippets related to data science, data analysis, data engineering, and Python programming.
Follow Us on Social Media
Copyright © 2025 Code Cut - All rights reserved.