获取NumPy数组的argsort

Question

我有一个如下的numpy数组：

array([ True,  True,  True,  True,  True, False,  True,  True, False,
        True, False,  True,  True,  True,  True,  True,  True, False,
       False, False, False, False,  True,  True, False, False, False,
        True,  True,  True,  True,  True,  True,  True, False,  True,
        True,  True,  True, False,  True,  True, False, False,  True,
        True,  True, False,  True,  True,  True, False], 

我想获取所有True元素的索引。在numpy中没有像Pandas Series那样的get_loc方法，也没有类似于列表的index方法。我不想将其转换为列表，然后使用.index方法。

有什么办法吗？

- Baktaawar

Answer 1

使用ndarray.nonzero：

>>> a.nonzero()
(array([ 0,  1,  2,  3,  4,  6,  7,  9, 11, 12, 13, 14, 15, 16, 22, 23, 27,
        28, 29, 30, 31, 32, 33, 35, 36, 37, 38, 40, 41, 44, 45, 46, 48, 49,
        50]),)

- wim

Answer 2

在pandas中实现这个功能：

In [255]:

s[s==True].index
Out[255]:
Int64Index([0, 1, 2, 3, 4, 6, 7, 9, 11, 12, 13, 14, 15, 16, 22, 23, 27, 28, 29, 30, 31, 32, 33, 35, 36, 37, 38, 40, 41, 44, 45, 46, 48, 49, 50], dtype='int64')

更新

实际上，您可以利用值已经是布尔值的事实来掩盖这个系列:

In [256]:

s[s].index
Out[256]:
Int64Index([0, 1, 2, 3, 4, 6, 7, 9, 11, 12, 13, 14, 15, 16, 22, 23, 27, 28, 29, 30, 31, 32, 33, 35, 36, 37, 38, 40, 41, 44, 45, 46, 48, 49, 50], dtype='int64')

同样地，对于numpy数组，您可以使用布尔值来掩盖数组，并使用np.where获取索引值：

In [261]:

np.where(a)
​
Out[261]:
(array([ 0,  1,  2,  3,  4,  6,  7,  9, 11, 12, 13, 14, 15, 16, 22, 23, 27,
        28, 29, 30, 31, 32, 33, 35, 36, 37, 38, 40, 41, 44, 45, 46, 48, 49,
        50], dtype=int64),)

- EdChum

Answer 3

np.ix_ 的方式似乎是最慢的。

In [846]: % timeit a.nonzero()
1000000 loops, best of 3: 707 ns per loop

In [845]: % timeit np.where(a)
1000000 loops, best of 3: 883 ns per loop

In [849]: %timeit np.ix_(a==True)
100000 loops, best of 3: 9.21 µs per loop

- fixxxer