numpy.dsplit#
- numpy.dsplit(ary, indices_or_sections)[source]#
沿着第三个轴(深度)将数组拆分为多个子数组。
请参考
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)]