Python NumPy split方法用法及代码示例

例子

基本用法

a = np.array([4,5,6,7,8,9])
np.split(a, 2)



[array([4, 5, 6]), array([7, 8, 9])]

当分割不均匀时，会抛出错误：

a = np.array([4,5,6,7,8])
np.split(a, 2)



ValueError: array split does not result in an equal division

通过切片分割

a = np.array([4,5,6,7,8,9])
np.split(a, (2,3))



[array([4, 5]), array([6]), array([7, 8, 9])]

在这里，我们按如下方式进行切片：

a[:2]
a[2:3]
a[3:]

分割二维数组

考虑以下二维数组：

a = np.array([[1,2,3,4],[5,6,7,8]])
a



array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

水平砍伐

按行拆分a：

np.split(a, 2, axis=0)



[array([[1, 2, 3, 4]]), array([[5, 6, 7, 8]])]

垂直砍伐

按列拆分a：

np.split(a, 2, axis=1)



[array([[1, 2],
        [5, 6]]),
 array([[3, 4],
        [7, 8]])]

切片

要通过切片按列分割a：

np.split(a, (2,3), axis=1)



[array([[1, 2],
        [5, 6]]),
 array([[3],
        [7]]),
 array([[4],
        [8]])]

在这里，我们获得的列是：

first two columns
third column
fourth column