numpy.dsplit#

numpy.dsplit(ary,indices_or_sections)[source]#

Split array into multiple sub-arrays along the 3rd axis (depth).

Please refer to thesplit documentation.dsplit is equivalenttosplit withaxis=2, the array is always split along the thirdaxis provided the array dimension is greater than or equal to 3.

See also

split

Split an array into multiple sub-arrays of equal size.

Examples

Try it in your browser!

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

Go BackOpen In Tab