python 数组分块 python数组分段

数组分割相关函数介绍

函数

数组及操作

split

将一个数组分割为多个子数组

hsplit

将一个数组水平分割为多个子数组(按列)

vsplit

将一个数组垂直分割为多个子数组(按行)

1)

沿特定的轴将数组分割为子数组

numpy.split(ary, indices_or_sections, axis)
ary：被分割的数组
indices_or_sections：果是一个整数，就用该数平均切分，如果是一个数组，为沿轴切分的位置(左开右闭)
axis：沿着哪个维度进行切向，默认为0，横向切分。为1时，纵向切分
>>> import numpy as np
>>> a = np.arange(9)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8])
>>> np.split(a,3)
[array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8])]
>>> np.split(a,4)
Traceback (most recent call last):
ValueError: array split does not result in an equal division

2)

用于水平分割数组，通过指定要返回的相同形状的数组数量来拆分原数组

>>> import numpy as np
>>> a = np.floor(10 * np.random.random((2, 6)))
>>> a
array([[1., 1., 5., 8., 6., 5.],
[1., 3., 1., 3., 7., 8.]])
>>> np.hsplit(a, 3)
[array([[1., 1.],
[1., 3.]]), array([[5., 8.],
[1., 3.]]), array([[6., 5.],
[7., 8.]])]

3)

沿着垂直轴分割，其分割方式与hsplit用法相同

>>> import numpy as np
>>> a = np.arange(16).reshape(4,4)
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15]])
>>> np.vsplit(a,2)
[array([[0, 1, 2, 3],
[4, 5, 6, 7]]), array([[ 8, 9, 10, 11],
[12, 13, 14, 15]])]