python的split分割数组_NumPy数组的分割

在 NumPy 中，利用 split()、hsplit() 和 vsplit() 等函数可实现数组的分割操作。

split() 函数

该函数可沿特定的轴将数组分割为子数组。使用 split() 函数的方法如下：

numpy.split(arr,indices_or_sections,axis)

其中，参数 arr 表示被分割的数组，indices_or_sections 表示从 arr 数组创建的大小相同的子数组的数量，可以为整数。如果此参数是一维数组，则该参数表示在 arr 数组中的分割点，arr 数组将按照分割点来分割数组。axis 表示返回数组中的轴，默认为 0，表示竖直方向分割，1 表示水平方向分割。

hsplit() 函数

该函数是 split() 函数的特例，它是将数组沿着水平方向分割，即将一个数组按列分割为多个子数组。使用 hsplit() 函数的方法如下：

numpy.hsplit(arr,indices_or_sections)

其中，参数 arr 表示被分割的数组，indices_or_sections 表示将 arr 数组创建为大小相同的子数组的数量。如果此参数是一维数组，则该参数表示在 arr 数组中的分割点，arr 数组将按照分割点来分割数组。

vsplit()函数

该函数是 split() 函数的特例，它是将数组沿着竖直方向分割，即将一个数组按行分割为多个子数组。使用 vsplit() 函数的方法如下：

numpy.vsplit(arr,indices_or_sections)

其中，参数 arr 表示被分割的数组；indices_or_sections 表示将 arr 数组创建为大小相同的子数组的数量。如果此参数是一维数组，则该参数表示在 arr 数组中的分割点，arr 数组将按照分割点来分割数组。

数组的分割示例

数组的分割示例代码 example1 如下。

# -*- coding: UTF-8 -*-

import numpy as np

arr1 = np.array([[1,2,3], [4,5,6]]) #创建数组arr1

print('第1个数组arr1：',arr1)

arr2 =np.arange(9) #创建数组arr2

print('第2个数组arr2：',arr2)

#使用split函数

print('将arr1数组竖直分割为2个大小相等的子数组：')

print (np.split(arr1,2))

print('将arr1数组水平分割为3个大小相等的子数组：')

print (np.split(arr1,3,1))

print('将arr2数组在一维数组中标明的位置分割：')

print (np.split(arr2, [2, 5]))

#使用hsplit函数和vsplit函数

print ('arr1数组水平分割：')

print(np.hsplit(arr1,3))

print ('arr1数组竖直分割：')

print(np.vsplit(arr1,2))

print ('arr2数组水平分割：')

print (np.hsplit(arr2, [2, 5]))