NumPy数组中元素的索引

Question

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

但是，当我尝试执行NumPy数组时：

decoding.index(i)

我得到：

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

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

Answer 1

使用np.where来获得，其中一个给定的条件是指数True。

例子：

对于np.ndarray称为的2D 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()方法创建的子类：

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)

测试：

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