Python numpy array_split用法及代碼示例

本文簡要介紹 python 語言中 numpy.array_split 的用法。

用法:

numpy.array_split(ary, indices_or_sections, axis=0)

將一個數組拆分為多個子數組。

請參閱split文檔。這些函數之間的唯一區別是array_split允許indices_or_sections是一個整數不是等分軸。對於長度為 l 的數組，應分為 n 個部分，它返回 l % n 個大小為 l//n + 1 的子數組以及其餘大小為 l//n 的子數組。

例子：

>>> x = np.arange(8.0)
>>> np.array_split(x, 3)
[array([0.,  1.,  2.]), array([3.,  4.,  5.]), array([6.,  7.])]

>>> x = np.arange(9)
>>> np.array_split(x, 4)
[array([0, 1, 2]), array([3, 4]), array([5, 6]), array([7, 8])]