在 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