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])