Python numpy delete用法及代碼示例

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

用法:

numpy.delete(arr, obj, axis=None)

返回一個新數組，其中刪除了沿軸的子數組。對於一維數組，這將返回那些未返回的條目arr[對象].

參數：

arr： array_like

輸入數組。

obj： 切片、int 或整數數組

指示要沿指定軸刪除的子數組的索引。

axis： 整數，可選

沿其刪除由 obj 定義的子數組的軸。如果axis為None，則將obj應用於展平數組。

返回：

out： ndarray

一份arr與指定的元素對象刪除。注意delete不會發生在原地。如果軸是無，out是一個扁平數組。

注意：

通常最好使用布爾掩碼。例如：

>>> arr = np.arange(12) + 1
>>> mask = np.ones(len(arr), dtype=bool)
>>> mask[[0,2,4]] = False
>>> result = arr[mask,...]

等效於 np.delete(arr, [0,2,4], axis=0)，但允許進一步使用掩碼。

例子：

>>> arr = np.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]])
>>> arr
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [ 9, 10, 11, 12]])
>>> np.delete(arr, 1, 0)
array([[ 1,  2,  3,  4],
       [ 9, 10, 11, 12]])

>>> np.delete(arr, np.s_[::2], 1)
array([[ 2,  4],
       [ 6,  8],
       [10, 12]])
>>> np.delete(arr, [1,3,5], None)
array([ 1,  3,  5,  7,  8,  9, 10, 11, 12])