numpy.hsplit() 函数

numpy.hsplit 是 split() 函数的一个特例,其中轴为 1 表示水平分割,而不管输入数组的维度如何。


示例

import numpy as np 
a = np.arange(16).reshape(4,4) 

print 'First array:' 
print a 
print '\n'  

print 'Horizontal splitting:' 
b = np.hsplit(a,2) 
print b 
print '\n'

它的输出将如下所示 −

First array:
[[ 0 1 2 3]
 [ 4 5 6 7]
 [ 8 9 10 11]
 [12 13 14 15]]

Horizontal splitting:                                                         
[array([[ 0,  1],                                                             
       [ 4,  5],                                                              
       [ 8,  9],                                                              
       [12, 13]]), array([[ 2,  3],                                           
       [ 6,  7],                                                              
       [10, 11],                                                              
       [14, 15]])] 

❮ NumPy 数组操作