NumPy数组的分割

NumPy数组的分割

方法	说明
split()	分割数组
hsplit()	沿横轴分割数组
vsplit()	沿纵轴分割数组

1.split()

np.split(arr,整数或数组,axis=n)

(1)整数

import numpy as np

arr = np.arange(10)
res = np.split(arr, 5)
print("arr:")
print(arr)
print("res:")
print(res)

填整数是就是将数组分为多少段

(2)数组

import numpy as np

arr = np.arange(10)
res = np.split(arr, [2, 3, 5])
print("arr:")
print(arr)
print("res:")
print(res)

2.hsplit() & vsplit()

hsplit() 和vsplit()就是split()在不同维度的操作，具体可以和我上一篇博客类比学习。