numpy删除指定的几个元素|极客笔记

numpy删除指定的几个元素

在使用numpy库处理数组时，有时候我们需要删除数组中的特定元素。这可以通过numpy库中的delete()函数来实现。在本文中，我们将详细介绍如何使用numpy删除数组中指定的几个元素。

`delete()`函数的用法

delete()函数的用法如下：

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

arr: 输入的数组
obj: 要删除的元素的索引或切片
axis: 沿着哪个轴删除元素，默认为None，表示展开到一个一维数组

示例

让我们通过一个示例来演示如何使用delete()函数删除数组中的特定元素。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
indices = [0, 2, 4]

new_arr = np.delete(arr, indices)

print(new_arr)

运行上面的代码，输出如下：

[2 4]

在上面的示例中，我们首先创建了一个包含元素[1, 2, 3, 4, 5]的numpy数组arr。然后，我们定义了一个包含要删除元素索引的列表indices，在这个示例中，我们要删除索引为0、2和4的元素。最后，我们调用np.delete(arr, indices)来删除这些元素，并将结果存储在new_arr中。

实际应用

除了删除数组中的特定元素外，delete()函数还可以用于删除数组中的行或列。让我们通过一个示例来演示如何删除数组中的行或列。

删除行

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

row_indices = [0, 2]

new_arr = np.delete(arr, row_indices, axis=0)

print(new_arr)

运行上面的代码，输出如下：

[[4 5 6]]

在上面的示例中，我们首先创建了一个包含3行3列的二维numpy数组arr。然后，我们定义了一个包含要删除行索引的列表row_indices，在这个示例中，我们要删除第一行和第三行。最后，我们调用np.delete(arr, row_indices, axis=0)来删除这些行，并将结果存储在new_arr中。

删除列

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

col_indices = [0, 2]

new_arr = np.delete(arr, col_indices, axis=1)

print(new_arr)

运行上面的代码，输出如下：

[[2]
 [5]
 [8]]

在上面的示例中，我们首先创建了一个包含3行3列的二维numpy数组arr。然后，我们定义了一个包含要删除列索引的列表col_indices，在这个示例中，我们要删除第一列和第三列。最后，我们调用np.delete(arr, col_indices, axis=1)来删除这些列，并将结果存储在new_arr中。