python - 从数组中删除元素

Question

我有一个 numpy 数组，我想删除数组的前 3 个元素。我试过这个解决方案:

           a = np.arange(0,10)
           i=0
           while(i<3):
             del a[0]
             i=i+1

这给我一个错误“ValueError:无法删除数组元素”。我不明白为什么会这样。非常感谢您的帮助!

Answer 1

Numpy 数组具有固定大小，因此您不能简单地从中删除一个元素。实现你想要的最简单的方法是使用切片:

a = a[3:]

这将创建一个从原始数组的第 4 个元素开始的新数组。

对于某些场景，切片是不够的。如果你想创建一个由原始数组中的特定元素组成的子数组，你可以使用另一个数组来选择索引:

>>> a = arange(10, 20)
>>> a[[1, 4, 5]]
array([11, 14, 15])

所以基本上，a[[1,4,5]] 将返回一个由原始数组的元素 1、4 和 5 组成的数组。