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