首页 > NumPy 阅读数:33

NumPy数组的分割

< 上一页NumPy数组的连接 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]))
< 上一页NumPy数组的连接 NumPy数组转换下一页 >