NumPy 数组中元素的索引[重复]

在Python中，我们可以使用以下方法获取数组中值的索引.index().

但是对于 NumPy 数组，当我尝试执行以下操作时：

decoding.index(i)

I get:

AttributeError：“numpy.ndarray”对象没有属性“index”

我如何在 NumPy 数组上执行此操作？

Use np.where获取给定条件的索引True.

例子：

对于 2Dnp.ndarray called a:

i, j = np.where(a == value) # when comparing arrays of integers

i, j = np.where(np.isclose(a, value)) # when comparing floating-point arrays

对于一维数组：

i, = np.where(a == value) # integers

i, = np.where(np.isclose(a, value)) # floating-point

请注意，这也适用于以下条件>=, <=, !=等等...

您还可以创建一个子类np.ndarray与index() method:

class myarray(np.ndarray):
    def __new__(cls, *args, **kwargs):
        return np.array(*args, **kwargs).view(myarray)
    def index(self, value):
        return np.where(self == value)

Testing:

a = myarray([1,2,3,4,4,4,5,6,4,4,4])
a.index(4)
#(array([ 3,  4,  5,  8,  9, 10]),)