Функция dsplit

Функция dsplit разделяет трехмерный массив по третьей оси (оси глубины) на несколько подмассивов. Функция принимает массив и количество частей или индексы, по которым нужно произвести разделение. Возвращается список массивов.

Синтаксис

np.dsplit(ary, indices_or_sections)

Пример

Разделим трехмерный массив размером 2 на 2 на 6 на 3 равные части по глубине:

import numpy as np

arr = np.arange(24).reshape(2, 2, 6)
res = np.dsplit(arr, 3)
for sub_arr in res:
	print(sub_arr)

Результат выполнения кода:

[[[0 1]
  [6 7]]

 [[12 13]
  [18 19]]]
[[[2 3]
  [8 9]]

 [[14 15]
  [20 21]]]
[[[4 5]
  [10 11]]

 [[16 17]
  [22 23]]]

Пример

Разделим массив по глубине на части с помощью указания индексов 2 и 4:

import numpy as np

arr = np.arange(24).reshape(2, 2, 6)
res = np.dsplit(arr, (2, 4))
for sub_arr in res:
	print(sub_arr)

Результат выполнения кода:

[[[0 1]
  [6 7]]

 [[12 13]
  [18 19]]]
[[[2 3]
  [8 9]]

 [[14 15]
  [20 21]]]
[[[4 5]
  [10 11]]

 [[16 17]
  [22 23]]]

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

функцию split,
которая разделяет массив по указанной оси
функцию hsplit,
которая разделяет массив по горизонтали
функцию vsplit,
которая разделяет массив по вертикали
функцию array_split,
которая разделяет массив на части с возможностью неравного деления