numpy.dsplit
numpy.dsplit(ary, indices_or_sections) [source]
沿第3轴(深度)将数组拆分为多个子数组。
请参split 分文档。dsplit等价于axis=2
进行split ,如果数组维数大于或等于3,则始终沿着第三个轴进行数组分割。
例子
>>> 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)]