新彩天欢迎您!
幻海优品

Python numpy.vsplit函数方法的使用

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

免责声明:以上内容(如有图片或视频亦包括在内)有转载其他网站资源,如有侵权请联系删除