Python numpy split用法及代码示例

本文简要介绍 python 语言中 numpy.split 的用法。

用法:

numpy.split(ary, indices_or_sections, axis=0)

将数组拆分为多个子数组作为 ary 中的视图。

参数：

ary： ndarray

数组被划分为子数组。

indices_or_sections： int 或一维数组

如果indices_or_sections是整数N，则数组将沿轴分为N个相等的数组。如果无法进行这样的拆分，则会引发错误。

如果indices_or_sections是排序整数的一维数组，条目指示沿轴数组被拆分。例如，[2, 3]会，因为axis=0， 造成

• ary[:2]

• 阿里[2:3]

• 阿里[3:]

如果索引超出了数组沿轴的维度，则相应地返回一个空的sub-array。

axis： 整数，可选

要分割的轴，默认为 0。

返回：

sub-arrays： ndarrays 列表

作为 ary 视图的子数组列表。

抛出：

ValueError

如果 indices_or_sections 以整数形式给出，但拆分不会导致等分。

例子：

>>> x = np.arange(9.0)
>>> np.split(x, 3)
[array([0.,  1.,  2.]), array([3.,  4.,  5.]), array([6.,  7.,  8.])]

>>> x = np.arange(8.0)
>>> np.split(x, [3, 5, 6, 10])
[array([0.,  1.,  2.]),
 array([3.,  4.]),
 array([5.]),
 array([6.,  7.]),
 array([], dtype=float64)]