Slice 2d array into smaller 2d arrays

Learn, how to slice 2d array into smaller 2d arrays in Python?
Submitted by Pranit Sharma, on January 24, 2023

NumPy is an abbreviated form of Numerical Python. It is used for different types of scientific operations in python. Numpy is a vast library in python which is used for almost every kind of scientific or mathematical operation. It is itself an array which is a collection of various methods and functions for processing the arrays.

Problem statement

Suppose that we are given a two-dimensional array with more columns and we need to split this NumPy array into smaller two-dimensional arrays with fewer columns.

Slicing 2d array into smaller 2d arrays

For this purpose, we will use the numpy.split() method where we will pass the total number of required sub-arrays that we want and the original array that has to be divided.

Let us understand with the help of an example,

Python code to slice 2d array into smaller 2d arrays

# Import numpy
import numpy as np

# Creating a numpy array
arr = np.arange(10).reshape([2,5])

# Display original array
print("Original array:\n",arr,"\n")

# Splitting array into 5 sub arrays
res = np.split(arr,5,axis=1)

# Display result


Example: Slice 2d array into smaller 2d arrays

In this example, we have used the following Python basic topics that you should learn:

Python NumPy Programs »

Comments and Discussions!

Load comments ↻

Copyright © 2024 All rights reserved.