删除数组中的索引

Question

我有以下代码用于删除数组中的索引，但它似乎不起作用

import numpy as np
length=4
indices=np.arange(length)
for num in (indices):
        np.delete(indices, num)
        print("checking", indices, num)

似乎是什么问题？ np.delete对 arrays 不起作用吗？

Answer 1

Numpy 数组是不可变的。 所以你不能从中删除一个项目。 但是，您可以构造一个没有您不想要的值的新数组，如下所示：

new_indices = np.delete(indices, [0,1,2])

在您的代码中，您可以尝试在删除元素后获取新数组，方法是将其分配给如下变量：

new_indices = np.delete(indices, num)
print(new_indices, indices) # prints [1,2,3],[0,1,2,3]

现在如果你打印 new_indices 它不会有索引 num 处的元素

Answer 2

numpy.delete返回给定数组的副本，它不会修改现有数组。

Answer 3

最好不要在更改容器长度的同时迭代容器。 尝试使用索引代替：

import numpy as np
length=4
indices=np.arange(length)
for i in reversed(range(len(indices))):
        indices = np.delete(indices, i)
        print("checking", indices, i)

如果您正在寻找按元素删除，您可以在使用删除之前找到元素索引：

import numpy as np
length=4
elements=np.array([1, 3, 2, 4])
for i in elements:
        idx = np.where(elements == i)
        elements = np.delete(elements, idx)
        print("checking", elements, i)