代码之家  ›  专栏  ›  技术社区  ›  Mauro Gentile

查找多个值上带有where子句的索引

  •  0
  • Mauro Gentile  · 技术社区  · 8 年前
    np.where(A ==2)[0]
    

    给出的索引 A 其中元素等于2。

    你如何归纳出一系列可能的值?

    我正在寻找类似以下内容:

    np.where(A in ([2,3,6,8]))[0]
    
    1 回复  |  直到 8 年前
        1
  •  1
  •   Stephen Rauch Afsar Ali    8 年前

    自从 NumPy 1.13 您可以使用 isin 作用

    在以前的版本中 in1d .

    测试:

    A = np.array([1, 2, 3, 4, 5])
    print(np.isin(A, [2, 3, 6, 8]))
    

    结果:

    [False  True  True False False]