bundles / numpy 2.4.3 / numpy / vsplit
_ArrayFunctionDispatcher
numpy:vsplit
Signature
def vsplit(ary, indices_or_sections)Summary
Split an array into multiple sub-arrays vertically (row-wise).
Extended Summary
Please refer to the split documentation. vsplit is equivalent to split with axis=0 (default), the array is always split along the first axis regardless of the array dimension.
Examples
import numpy as np x = np.arange(16.0).reshape(4, 4)✓
x np.vsplit(x, 2) np.vsplit(x, np.array([3, 6]))✗
[array([[ 0., 1., 2., 3.],
[ 4., 5., 6., 7.],
[ 8., 9., 10., 11.]]), array([[12., 13., 14., 15.]]), array([], shape=(0, 4), dtype=float64)]
With a higher dimensional array the split is still along the first axis.
x = np.arange(8.0).reshape(2, 2, 2)
✓x np.vsplit(x, 2)✗
[array([[[0., 1.],
[2., 3.]]]), array([[[4., 5.],
[6., 7.]]])]
See also
- split
Split an array into multiple sub-arrays of equal size.
Aliases
numpy.vsplit