numpy.split() in Python
The numpy.split() function splits an array into multiple sub-arrays.
Syntax
numpy.split(ary, indices_or_sections, axis=0)
Parameter
ary: This parameter represents the Array to be divided into sub-arrays.
indices_or_sections : This parameter represents an int or an 1-D array. If indices_or_sections is an integer, N, the array will be divided into N equal arrays along axis. If such a split is not possible, an error is raised.
axis: This parameter represents an axis along which the array is split. It is an optional parameter which takes int values, and by default, it is 0.
Return
This function returns a list of sub-arrays.
Raises
This function raises a ValueError if the parameter ‘indices_or_sections’ is given as an integer, but a split does not result in equal division.
Example 1
# Python Program explaining
# numpy.split() function
import numpy as np
inp_arr = np.arange(9)
print ('First array:' )
print(inp_arr,'\n')
print ('Split the array in 3 equal-sized subarrays:')
out = np.split(inp_arr,3)
print (out)
Output
First array:
[0 1 2 3 4 5 6 7 8]
Split the array in 3 equal-sized subarrays:
[array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8])]