numpy.dsplit() function : This function Split array into multiple sub-arrays along the 3rd axis (depth).
Syntax:numpy.dsplit(ary, indices_or_sections)
#example program
import numpy as np
x=np.array([[1,2,3],[4,5,6],[7,8,9]])
y=np.dsplit(x, 1)
print(y)