Python数据分析三大框架之numpy （二）分割

前段时间学习了分割数组的几种方法

 

• 按列分割 axis = 1

import numpy as np

A = np.arange(12).reshape((3, 4))
print(A)

# 按列分割成两块 按列分割 axis = 1
print(np.split(A, 2, axis= 1))


# 按行分割成三块
print(np.vsplit(A, 3))



#result
[[ 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]])]

 

• 按行分割， axis = 0

# 按行分割成三块 按行分割 axis = 0
print(np.split(A, 3, axis= 0))

# 按列分割成两块
print(np.hsplit(A, 2))



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

 

 

 

• 不完全的分割

 

# 不等的分割,按列来分割成三块
print(np.array_split(A, 3, axis=1))

 

又是一部分这个课