How to split an array in Python?

[SubhanUllah]

To split an array, you can import numpy and use the

np.array_split(array, # of splits) 

method.

for example:

arr = np.array([1, 2, 3, 4, 5, 6]) 
newarr = np.array_split(arr, 3) 
print(newarr) 
 
>> [array([1, 2]), array([3, 4]), array([5, 6])]