numpy 获取特定数值的索引,如何从NumPy数组中获取除特定索引之外的所有值？

I have a NumPy array, and I want to retrieve all the elements except a certain index. For example, consider the following array

a = [0,1,2,3,4,5,5,6,7,8,9]

If I specify index 3, then the resultant should be

a = [0,1,2,4,5,5,6,7,8,9]

解决方案

Like resizing, removing elements from an NumPy array is a slow operation (especially for large arrays since it requires allocating space and copying all the data from the original array to the new array).

It should be avoided if possible.

Often you can avoid it by working with a masked array instead. For example, consider the array a:

import numpy as np

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

print(a)

print(a.sum())

# [0 1 2 3 4 5 5 6 7 8 9]

# 50

We can mask its value at index 3 and can perform a summation which ignores masked elements:

a = np.ma.array(a, mask=False)

a.mask[3] = True

print(a)

print(a.sum())

# [0 1 2 -- 4 5 5 6 7 8 9]

# 47

Masked arrays also support many operations besides sum.

If you really need to, it is also possible to remove masked elements using the compressed method:

print(a.compressed())

# [0 1 2 4 5 5 6 7 8 9]

But as mentioned above, avoid it if possible.