numpy数组删除某个元素

要删除numpy数组中的某个元素，可以使用numpy.delete()函数。该函数的参数包括要操作的数组、要删除的元素的索引或切片以及要删除的轴。具体操作如下所示：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.delete(arr, 2)  # 删除索引为2的元素
print(new_arr)  # 输出 [1, 2, 4, 5]

在上述示例中，我们创建了一个包含1到5的numpy数组。然后，使用np.delete()函数删除了索引为2的元素，即数组中的数字3。最后，打印出新的数组new_arr，可以看到数字3已经被成功删除了。

如果要删除多个元素，可以将要删除的索引或切片作为一个列表传递给np.delete()函数的第二个参数。例如：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.delete(arr, [1, 3])  # 删除索引为1和3的元素
print(new_arr)  # 输出 [1, 3, 5]

在上述示例中，我们删除了索引为1和3的元素，即数组中的数字2和4。最后，打印出新的数组new_arr，可以看到数字2和4已经被成功删除了。

需要注意的是，np.delete()函数返回的是一个新的数组，原始数组并没有被修改。如果想要在原始数组上进行修改，可以直接对原始数组进行赋值操作。