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]])]