Python numpy.dsplit函数方法的使用

numpy.dsplit

numpy.dsplit(ary, indices_or_sections)      [source]

沿第3轴（深度）将数组拆分为多个子数组。

请参split 分文档。dsplit等价于axis=2进行split ，如果数组维数大于或等于3，则始终沿着第三个轴进行数组分割。

例子

1）按数量分割

import numpy as np

# 创建一个形状为 (3, 4, 6) 的三维数组
arr = np.arange(36).reshape(3, 4, 6)

# 将数组按深度轴分成 3 个子数组
split_arr = np.dsplit(arr, 3)

# 输出每个子数组的形状
for sub_arr in split_arr:
    print(sub_arr.shape)

2）按指定切割点分割

import numpy as np

# 创建一个形状为 (3, 4, 6) 的三维数组
arr = np.arange(36).reshape(3, 4, 6)

# 按照深度轴的指定索引进行分割
split_arr = np.dsplit(arr, [2, 4])

# 输出每个子数组的形状
for sub_arr in split_arr:
    print(sub_arr.shape)