numpy.vsplit
numpy.vsplit(ary, indices_or_sections) [source]
垂直(行)将数组拆分为多个子数组。
请参考split文档。vsplit
等价于axis=0
(默认)进行split,无论数组的维度如何,数组总是沿着第一个轴进行分割。
例子
>>> x = np.arange(16.0).reshape(4, 4)>>> xarray([[ 0., 1., 2., 3.], [ 4., 5., 6., 7.], [ 8., 9., 10., 11.], [12., 13., 14., 15.]])>>> np.vsplit(x, 2)[array([[0., 1., 2., 3.], [4., 5., 6., 7.]]), array([[ 8., 9., 10., 11.], [12., 13., 14., 15.]])]>>> 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)]
对于高维数组,分割仍沿第一轴
>>> x = np.arange(8.0).reshape(2, 2, 2)>>> xarray([[[0., 1.], [2., 3.]], [[4., 5.], [6., 7.]]])>>> np.vsplit(x, 2)[array([[[0., 1.], [2., 3.]]]), array([[[4., 5.], [6., 7.]]])]