Python numpy split用法及代碼示例

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

用法:

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

將數組拆分為多個子數組作為 ary 中的視圖。

參數：

ary： ndarray

數組被劃分為子數組。

indices_or_sections： int 或一維數組

如果indices_or_sections是整數N，則數組將沿軸分為N個相等的數組。如果無法進行這樣的拆分，則會引發錯誤。

如果indices_or_sections是排序整數的一維數組，條目指示沿軸數組被拆分。例如，[2, 3]會，因為axis=0， 造成

• ary[:2]

• 阿裏[2:3]

• 阿裏[3:]

如果索引超出了數組沿軸的維度，則相應地返回一個空的sub-array。

axis： 整數，可選

要分割的軸，默認為 0。

返回：

sub-arrays： ndarrays 列表

作為 ary 視圖的子數組列表。

拋出：

ValueError

如果 indices_or_sections 以整數形式給出，但拆分不會導致等分。

例子：

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

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