Numpy——数组分割

1.纵向等量分割

import numpy as np
A = np.arange(1,25).reshape((4,6))

print(A)
print(np.split(A, 3 ,axis = 1))  #纵向等分3列(第二个参数一定要能被总列数整除)

2.横向等量分割

import numpy as np
A = np.arange(1,25).reshape((4,6))

print(A)
print(np.split(A, 2, axis = 0))   #横向等分2行(第二个参数一定要能被总行数整除)

3.纵向不等量的分割

在机器学习时经常会需要将数据做不等量的分割，因此解决办法为np.array_split()

import numpy as np
A = np.arange(1,25).reshape((4,6))

print(A)
print(np.array_split(A,4,axis = 1))  #6列分成了2、2、1、1

4.横向不等量的分割

print(np.array_split(A,3,axis = 0))  #4行分成了2、1、1

5.vsplit和hsplit

print(np.vsplit(A,2))  #等价于print(np.split(A, 2, axis = 0))

print(np.hsplit(A,3))  #等价于print(np.split(A, 3 ,axis = 1))