Indexing numpy array with another numpy array

后端 未结 2 560
说谎
说谎 2020-12-15 04:16

Suppose I have

a = array([[1, 2],
           [3, 4]])

and

b = array([1,1])

I\'d like to use b in index a

2条回答
  •  余生分开走
    2020-12-15 05:00

    According the numpy tutorial:

    a[tuple(b)] 
    

    is the correct way to do this:

    http://www.scipy.org/Tentative_NumPy_Tutorial#head-3f4d28139e045a442f78c5218c379af64c2c8c9e

提交回复
热议问题