开发手册 欢迎您!
软件开发者资料库

Python numpy.dsplit函数方法的使用

NumPy(Numerical Python的缩写)是一个开源的Python科学计算库。使用NumPy,就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数,涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中dsplit方法的使用。

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)]