极客教程numpy delete详解
在NumPy库中，numpy.delete函数用于删除数组中指定位置的元素。本文将详细介绍如何使用numpy.delete函数以及一些常见的用法示例。接下来让我们一起来深入探讨吧。
语法
numpy.delete(arr, obj, axis=None)
arr: 输入的数组obj: 要删除的位置或者位置的序列axis: 沿着指定轴删除元素
示例1：删除一维数组中指定位置的元素
首先让我们看一个简单的示例，删除一维数组中指定位置的元素。
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.delete(arr, 2)
print(new_arr)
运行结果：
[1 2 4 5]
在这个示例中，我们定义了一个包含5个元素的一维数组arr，然后使用numpy.delete函数删除了位置为2的元素，即数组中的第3个元素。删除后的新数组new_arr为[1, 2, 4, 5]。
示例2：删除二维数组中指定行或列
接下来，让我们看一个删除二维数组中指定行或列的示例。
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_arr_row = np.delete(arr, 1, axis=0)
new_arr_col = np.delete(arr, 1, axis=1)
print("删除指定行的结果：")
print(new_arr_row)
print("删除指定列的结果：")
print(new_arr_col)
运行结果：
删除指定行的结果：
[[1 2 3]
[7 8 9]]
删除指定列的结果：
[[1 3]
[4 6]
[7 9]]
在这个示例中，我们定义了一个包含3行3列的二维数组arr，分别删除了第2行和第2列的元素。new_arr_row为删除指定行后的结果，new_arr_col为删除指定列后的结果。
示例3：删除多个位置的元素
除了可以删除单个位置的元素，numpy.delete函数还可以删除多个位置的元素。让我们看一个示例。
arr = np.array([1, 2, 3, 4, 5])
indices = [0, 2, 4]
new_arr = np.delete(arr, indices)
print(new_arr)
运行结果：
[2 4]
在这个示例中，我们定义了一个包含5个元素的一维数组arr，并指定了要删除的位置为0、2和4。numpy.delete函数会删除这些位置的元素，最终的结果为[2, 4]。
总结
通过以上示例，我们可以看到numpy.delete函数在处理数组时非常灵活，可以删除一维或多维数组中的指定位置的元素。在实际应用中，我们可以根据需求灵活运用该函数，对数组进行删除操作。