使用NumPy获取符合条件的索引列表

Question

我试图获取满足numpy数组中条件的元素的索引。例如：

import numpy as np
ar = np.array([1, 1, 2, 1, 3, 4, 1])
np.where(ar==1)[0]

将返回：

[0, 1, 3, 6]

现在，如果我希望条件为值列表，例如：

np.where(ar in [1, 2])[0]

我希望结果是：

[0, 1, 2, 3, 6]

如何使用numpy做到这一点？ 当然，我可以遍历条件列表并连接结果

indices = np.concatenate([np.where(ar==cond)[0] for cond in [1, 2])

但是我想知道是否还有更直接的类似numpy的方式。