numpy.dsplit(ary, indices_or_sections)
Split array into multiple sub-arrays along the 3rd axis (depth).
Please refer to the
documentation.
is equivalent to
with axis=2, the array is always split along the third axis provided the array dimension is greater than or equal to 3.
See also
Split an array into multiple sub-arrays of equal size.
Examples
>>> importnumpyasnp>>> x=np.arange(16.0).reshape(2,2,4)>>> xarray([[[ 0., 1., 2., 3.], [ 4., 5., 6., 7.]], [[ 8., 9., 10., 11.], [12., 13., 14., 15.]]])>>> np.dsplit(x,2)[array([[[ 0., 1.], [ 4., 5.]], [[ 8., 9.], [12., 13.]]]), array([[[ 2., 3.], [ 6., 7.]], [[10., 11.], [14., 15.]]])]>>> np.dsplit(x,np.array([3,6]))[array([[[ 0., 1., 2.], [ 4., 5., 6.]], [[ 8., 9., 10.], [12., 13., 14.]]]), array([[[ 3.], [ 7.]], [[11.], [15.]]]),array([], shape=(2, 2, 0), dtype=float64)]