分割数组

一、split()函数

1、该函数指沿指定的轴分割多个数组

numpy.split(ary,indices_or_sections,axis)

2、ary是要被分割的数组

3、indices_or_sections是一个整数或数组，如果是整数就用该数平均分割；如果是数组，则为沿指定轴的切片操作

4、axis指轴的分割方向，默认为0轴。

二、vsplit()函数

1、该函数指沿垂直方向分割数组，相当于split()函数axis=0情况

numpy.vsplit(ary,indices_or_sections)

2、ary是要被分割的数组

3、indices_or_sections是一个整数或数组，如果是整数就用该数平均分割；如果是数组，则为沿指定轴的切片操作

三、hsplit()函数

1、该函数指沿水平方向分割数组，相当于split()函数axis=1情况

numpy.hsplit(ary,indices_or_sections)

2、ary是要被分割的数组

3、indices_or_sections是一个整数或数组，如果是整数就用该数平均分

四、代码练习

# %% split()函数分割一维数组（整数）

import numpy as np

a1=np.arange(9)

b1=np.split(a1, 3)

print(b1)

print("--------------------------")

# %% split()函数分割一维数组（数组）

a1 = np.arange(9)

sections=np.array([4,7])

b1 = np.split(a1,sections)

print(b1)

print("--------------------------")

# %% split()函数和vsplit()函数分割二维数组（整数）

a2=np.array([[1,2,3,4],

             [5,6,7,8],

             [9,10,11,12],

             [13,14,15,16]])

b2=np.split(a2, 4)

print(b2)

print("--------------------------")

b3 = np.vsplit(a2, 4)

print(b3)

# %% split()函数和hsplit()函数分割二维数组（数组）

a2=np.array([[1,2,3,4],

             [5,6,7,8],

             [9,10,11,12],

             [13,14,15,16]])

sections=np.array([1,2])

b2=np.split(a2,sections,axis=1)

print(b2)

print("--------------------------")

b3=np.hsplit(a2,sections)

print(b3)

fffff