查找numpy数组中非X或Y的某个值的首次出现的索引

Question

我有一个如下形式的向量：

import numpy as np
vec = np.array([2, 2, 2, 51, 51, 52, 52, 14, 14, 14, 51, 51, 52, 52])

是否有一种简单的方法可以找到不是（例如）51或52的值的首次出现的索引？换句话说，返回以下索引的函数：[0，7]，其中0是2的第一个幻影的索引，7是14的第一个幻影的索引

Answer 1

import numpy as np
vec = np.array([2, 2, 2, 51, 51, 52, 52, 14, 14, 14, 51, 51, 52, 52])

first_occurrence = []
for x in np.unique(vec):
    if x not in [51,52]:
        first_occurrence.append(np.argmax(x==vec))

^{}查找布尔数组x==vec中第一次出现的最大值（即True）的索引。由于x来自vec，因此保证至少有一个True值

---

性能取决于vec的大小和要查找的值的数量。对于较大的数组，这种简单循环方法（蓝色）优于accepted answer（绿色和橙色），特别是对于在示例中找到的少量值（对于给定的玩具示例，它实际上快了1.7倍）（source）。 事实证明，将unique与index=True一起使用相对较慢，较大数组的另一个因素是掩码的内存分配