数组分割相关函数介绍
| 函数 | 数组及操作 |
|---|---|
| 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]])]
