numpy.array_split()
# welcome to softhunt.net
# import numpy
import numpy as np
array = np.arange(16)
# using numpy.array_split() method
softhunt = np.array_split(array, 6)
print(softhunt)
# welcome to softhunt.net
# import numpy
import numpy as np
array = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
# using numpy.array_split() method
softhunt = np.array_split(array, 3)
print(softhunt)