numpy科学计算器库入门7 array分割

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zy_505775013/article/details/88670224

import numpy as np
arr1 = np.arange(12).reshape((3,4))
print(arr1)

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

arr2,arr3 = np.split(arr1,2,axis=1)#水平方向分割 分成两份 等分
print(arr2)
print(arr3)

[[0 1]
 [4 5]
 [8 9]]
[[ 2  3]
 [ 6  7]
 [10 11]]

arr4,arr5,arr6 = np.split(arr1,3,axis=0)#垂直方向分割 分成3份 等分
print(arr4)
print(arr5)
print(arr6)

[[0 1 2 3]]
[[4 5 6 7]]
[[ 8  9 10 11]]

arr4,arr5,arr6 = np.array_split(arr1,3,axis=1)#水平方向分割 分成3份 不等分割
print(arr4)
print(arr5)
print(arr6)

[[0 1]
 [4 5]
 [8 9]]
[[ 2]
 [ 6]
 [10]]
[[ 3]
 [ 7]
 [11]]

arr5,arr6 = np.hsplit(arr1,2)#水平方向分割 分成3份 等分割
print(arr5)
print(arr6)

[[0 1]
 [4 5]
 [8 9]]
[[ 2  3]
 [ 6  7]
 [10 11]]

arr4,arr5,arr6 = np.split(arr1,3)#垂直方向分割 分成3份 等分割
print(arr4)
print(arr5)
print(arr6)

[[0 1 2 3]]
[[4 5 6 7]]
[[ 8  9 10 11]]