numpy.hsplit

numpy.hsplit(ary, indices_or_sections)

Split array into multiple sub-arrays horizontally.

Please refer to the numpy.split documentation. hsplit is equivalent to numpy.split with axis = 1.

See also

split

Split array into multiple sub-arrays.

Examples

>>> x = np.arange(16.0).reshape(4, 4)
>>> np.hsplit(x, 2)
<BLANKLINE>
[array([[  0.,   1.],
       [  4.,   5.],
       [  8.,   9.],
       [ 12.,  13.]]),
 array([[  2.,   3.],
       [  6.,   7.],
       [ 10.,  11.],
       [ 14.,  15.]])]

>>> np.hsplit(x, array([3, 6]))
<BLANKLINE>
[array([[  0.,   1.,   2.],
       [  4.,   5.,   6.],
       [  8.,   9.,  10.],
       [ 12.,  13.,  14.]]),
 array([[  3.],
       [  7.],
       [ 11.],
       [ 15.]]),
 array([], dtype=float64)]