笔记内容来自up主莫烦
https://www.bilibili.com/video/BV1Ex411L7oT?p=9
import numpy as np
"""
[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
1. 平均从行方向或者列方向 分割
1） 从行方向分成三个array
np.split(a,3,axis=0)
np.vsplit(a,3)
[array([[0, 1, 2, 3]]), array([[4, 5, 6, 7]]), array([[ 8, 9, 10, 11]])]
2）从列方向分成两部分array
np.split(a,2,axis=1)
np.hsplit(a,2)
[array([[0, 1],
[4, 5],
[8, 9]]), array([[ 2, 3],
[ 6, 7],
[10, 11]])]
2. 不平均分割 np.array_split()
把四列分成 2 1 1
print((np.array_split(a,3,axis=1)))
[array([[0, 1],
[4, 5],
[8, 9]]), array([[ 2],
[ 6],
[10]]), array([[ 3],
[ 7],
[11]])]
"""
a = np.arange(12).reshape((3,4))
print(a)
# 把4列分成2块
print(np.split(a,2,axis=1))
# 把3行分成3块
print(np.split(a,3,axis=0))
# array不等量分割
print((np.array_split(a,3,axis=1)))
print(np.vsplit(a,3))
print(np.hsplit(a,2))