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

numpy.split

numpy.split(ary, indices_or_sections, axis=0)     [source]

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

参数 :

aryndarray

数组可分为子数组。

indices_or_sections :int 或 1-D array

如果indexs_or_sections是整数N,

则该数组将沿轴分为N个相等的数组。

 如果无法进行此类拆分,则会引发错误。 

如果indexs_or_sections是一维排序的整数数组,

则条目指示沿轴在哪里拆分该数组。 

例如,对于axis = 0[2,3]

将导致 

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

如果索引沿轴超出数组的维数,

则将相应返回一个空的子数组。

axisint, 可选

沿其分割的轴,默认为0

返回值 :

sub-arrays :ndarrays的list

子数组列表,以查看ary。

Raises :

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

推荐文档

相关文档

大家感兴趣的内容

随机列表