python np.split函数

  • Post category:Python

当我们在处理NumPy数组时,常常需要将一个数组按指定的切分点分割成多个小数组,这时候可以使用NumPy的np.split函数。本篇攻略将对np.split函数进行详细讲解。

1.函数定义

np.split函数的定义如下:

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

2.函数参数

  • ary: 待切分的数组;
  • indices_or_sections: 所要划分的段数,整数或者一维数组形式,如果是整数n,则将数组分为n个等份;如果是一维数组,则按该数组给定的划分点分成多个段。需要注意的是,指定具体的分割点时,将会产生N-1个小数组,其中N为划分点的长度;
  • axis: 沿着哪个轴进行切分,默认为0。

3.函数返回值

返回切分后的数组列表。

4.函数示例

4.1 将数组分成n个等份

将数组按等份分成3份,示例代码如下:

import numpy as np

arr = np.array([1,2,3,4,5,6,7,8,9])

result = np.split(arr, 3)

print(result)

执行结果为:

[array([1, 2, 3]), array([4, 5, 6]), array([7, 8, 9])]

可以看到,数组已经被成功地分成了3份。

4.2 指定划分点进行数组分割

示例代码如下:

import numpy as np

arr = np.array([1,2,3,4,5,6,7,8,9])

result = np.split(arr, [2,5,7])

print(result)

执行结果为:

[array([1, 2]), array([3, 4, 5]), array([6, 7]), array([8, 9])]

可以看到,将数组按[2,5,7]指定的分割点分成了4份。其中,分割点2,5,7作为左闭右开区间的结尾点,将数组分成了4个小数组。

5.总结

通过上面的讲解,我们可以看到np.split函数的参数、返回值以及使用方法。希望这篇攻略能对大家使用np.split函数处理NumPy数组提供帮助。