当前位置: 首页>>代码示例>>用法及示例精选>>正文


python numpy split用法及代码示例

用法:

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

将一个数组拆分为多个子数组。

参数:
ary ndarray

数组可分为子数组。

indices_or_sections int 或 1-D array

如果indexs_or_sections是整数N,则该数组将沿轴划分为N个相等的数组。如果无法进行这种拆分,则会引发错误。

如果indexs_or_sections是一维排序的整数数组,则条目指示沿轴在哪里拆分该数组。例如,[2, 3]会,因为axis=0, 导致

  • ary[:2]
  • ary[2:3]
  • ary[3:]

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

axis int, 可选参数

沿其分割的轴,默认值为0。

返回值:
sub-arrays ndarrays列表

子数组列表。

异常:
ValueError

如果indexs_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)]

源码:

numpy.split的API实现见:[源代码]

相关用法

注:本文由纯净天空筛选整理自 numpy.split。非经特殊声明,原始代码版权归原作者所有,本译文的传播和使用请遵循“署名-相同方式共享 4.0 国际 (CC BY-SA 4.0)”协议。