nlcpy.hsplit

nlcpy.hsplit(ary, indices_or_sections)

Splits an array into multiple sub-arrays horizontally (column-wise).

Please refer to the split() documentation. hsplit is equivalent to split() with axis=1, the array is always split along the second axis regardless of the array dimension.

参考

split

Splits an array into multiple sub-arrays.

Examples

>>> import nlcpy as vp
>>> x = vp.arange(16.0).reshape(4, 4)
>>> x
array([[ 0.,  1.,  2.,  3.],
       [ 4.,  5.,  6.,  7.],
       [ 8.,  9., 10., 11.],
       [12., 13., 14., 15.]])
>>> vp.hsplit(x, 2)
[array([[ 0.,  1.],
       [ 4.,  5.],
       [ 8.,  9.],
       [12., 13.]]), array([[ 2.,  3.],
       [ 6.,  7.],
       [10., 11.],
       [14., 15.]])]
>>> vp.hsplit(x, vp.array([3, 6]))
[array([[ 0.,  1.,  2.],
       [ 4.,  5.,  6.],
       [ 8.,  9., 10.],
       [12., 13., 14.]]), array([[ 3.],
       [ 7.],
       [11.],
       [15.]]), array([], shape=(4, 0), dtype=float64)]