我试图获取满足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的方式。