Open In App

Python | numpy.array_split() method

Last Updated : 17 Sep, 2019
Suggest changes
Share
Like Article
Like
Report
With the help of numpy.array_split() method, we can get the splitted array of having different dimensions by using numpy.array_split() method.
Syntax : numpy.array_split() Return : Return the splitted array of one dimension.
Example #1 : In this example we can see that by using numpy.array_split() method, we are able to split the array in the number of subarrays by passing it as a parameter. Python3 1=1
# import numpy import numpy as np array = np.arange(9) # using numpy.array_split() method gfg = np.array_split(array, 4) print(gfg) 
Output :
[array([0, 1, 2]), array([3, 4]), array([5, 6]), array([7, 8])]
Example #2 : Python3 1=1
# import numpy import numpy as np array = [[1, 2, 3],  [4, 5, 6],  [7, 8, 9]] # using numpy.array_split() method gfg = np.array_split(array, 3) print(gfg) 
Output :
[array([[1, 2, 3]]), array([[4, 5, 6]]), array([[7, 8, 9]])]

Explore