Home »
Python »
Python Programs
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
print("Result:\n",res,"\n")
Output
In this example, we have used the following Python basic topics that you should learn:
Python NumPy Programs »