Python NumPy array_split方法用法及代碼示例

NumPy 的 array_split(~) 方法根據指定的參數劃分輸入數組。

兩個array_split(~) 和 split(~) 的參數和函數完全相同。主要區別在於，array_split(~) 允許在第二個參數為 int 的情況下進行不等除法，而 split(~) 方法會引發錯誤。

對於 array_split(~) ，最後一個分區將保存較少的數據以適應不等分。

1. a | array-like

要拆分的輸入數組。

2. indices_or_sections | int

如果給定 int n，則輸入數組將沿指定軸分成 n 個相等的數組。為了更準確地了解在哪裏分割數組，請提供一個已排序整數的一維數組。

3. axis | None 或 int | optional

執行分割所沿的軸。默認情況下，axis=0 。

包含分割的 NumPy 數組的列表。

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



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

當分割不均勻時，不會拋出錯誤：

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



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

a = np.array([4,5,6,7,8,9])
np.array_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.array_split(a, 2, axis=0)



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

按列拆分a：

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



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

要通過切片按列分割a：

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



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

在這裏，我們獲得的列是：

first two columns
third column
fourth column

相關用法

注：本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 NumPy | array_split method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。