numpy pandas基础知识笔记 4.array的分割

使用np.split对array进行一个分割

例：

import numpy as np
A = np.arange(12).reshape(3,4)
print(A)
print(np.split(A,2,axis=1))//对A列方向分割为两个array

输出:

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
[array([[0, 1],
       [4, 5],
       [8, 9]]), array([[ 2,  3],
       [ 6,  7],
       [10, 11]])]

注：不能进行不等量的分割

 

进行不等量的分割使用array_split

例：

import numpy as np
A = np.arange(12).reshape(3,4)
print(A)
print(np.array_split(A,3,axis=1))

输出：

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
[array([[0, 1],
       [4, 5],
       [8, 9]]), array([[ 2],
       [ 6],
       [10]]), array([[ 3],
       [ 7],
       [11]])]

 

vsplit纵向分割

hsplit横向分割