Indexing NumPy 2D array with another 2D array

后端 未结 6 1761
孤独总比滥情好
孤独总比滥情好 2021-01-05 15:16

I have something like

m = array([[1, 2],
            [4, 5],
            [7, 8],
            [6, 2]])

and

select = array([         


        
6条回答
  •  傲寒
    傲寒 (楼主)
    2021-01-05 15:42

    IMHO, this is simplest variant:

    m[np.arange(4), select]
    

提交回复
热议问题