Indexing NumPy 2D array with another 2D array

后端 未结 6 1747
孤独总比滥情好
孤独总比滥情好 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:43

    result = array([m[j][0] if i==0 else m[j][1] for i,j in zip(select, range(0, len(m)))])
    

提交回复
热议问题