numpy.split() in Python

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])]