代码之家  ›  专栏  ›  技术社区  ›  Nivi

使用布尔向量索引numpy数组

  •  1
  • Nivi  · 技术社区  · 8 年前

    我有一个numpy。ndarray和我有一个布尔列表。我想使用列表访问数组中的列。

    X = [[1,2,3,4],[5,6,7,8]]
    Y = [True,False,False,True]
    

    我希望结果是

    [[1,4][5,8]]
    

    我想我这样做效率很低,我想知道是否有一个简单的方法。

    1 回复  |  直到 8 年前
        1
  •  2
  •   utengr    8 年前

    你必须先把它转换成numpy。

    import numpy as np
    
    X = np.array([[1,2,3,4],[5,6,7,8]])
    Y = np.array([True,False,False,True])
    
    print(X[:,Y])