数组拆分

目录

1.横向拆分 hsplit()

2.纵向拆分 vsplit()

3.split()

---

1.横向拆分 hsplit()

import nmupy as np
arr=np.arange(16).reshape(4,4)
print('arr:',arr)

#横向拆分  平分
arr_new=np.hsplit(arr,2)
print('arr_new:',arr_new)

arr1,arr2=np.hsplit(arr,2)
print('arr1:',arr1)
print('arr2：',arr2)

---

2.纵向拆分 vsplit()

arr_new=np.vsplit(arr,2)
print('arr_new:',arr_new)

arr1,arr2=np.vsplit(arr,2)
print('arr1:',arr1)
print('arr2:',arr2)

---

3.split()

# 横向拆分
arr_new=np.split(arr,2,axis=1)
print('arr_new:',arr_new)
# 纵向拆分
arr_new=np.split(arr,2,axis=0)
print('arr_new:',arr_new)