numpy.dsplit

numpy.dsplit(ary, indices_or_sections)

Разделить массив на несколько подмассивов вдоль третьей оси (глубины).

Пожалуйста, обратитесь к split документация. dsplit эквивалентно split с axis=2, массив всегда разделяется по третьей оси, при условии, что размерность массива больше или равна 3.

Смотрите также

split

Разделить массив на несколько подмассивов одинакового размера.

Примеры

Попробуйте в вашем браузере!

>>> import numpy as np
>>> x = np.arange(16.0).reshape(2, 2, 4)
>>> x
array([[[ 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)]

НазадОткрыть во вкладке