[Numpy-discussion] indexing question

John Salvatier jsalvati@u.washington....
Sun Nov 21 13:28:56 CST 2010


yes use the symbol ':'

so you want

t[:,x,y]

2010/11/21 Ernest Adrogué <eadrogue@gmx.net>:
> Hi,
>
> Suppose an array of shape (N,2,2), that is N arrays of
> shape (2,2). I want to select an element (x,y) from each one
> of the subarrays, so I get a 1-dimensional array of length
> N. For instance:
>
> In [228]: t=np.arange(8).reshape(2,2,2)
>
> In [229]: t
> Out[229]:
> array([[[0, 1],
>        [2, 3]],
>
>       [[4, 5],
>        [6, 7]]])
>
> In [230]: x=[0,1]
>
> In [231]: y=[1,1]
>
> In [232]: t[[0,1],x,y]
> Out[232]: array([1, 7])
>
> This way, I get the elements (0,1) and (1,1) which is what
> I wanted. The question is: is it possible to omit the [0,1]
> in the index?
>
> Thanks in advance.
>
> --
> Ernest
> _______________________________________________
> NumPy-Discussion mailing list
> NumPy-Discussion@scipy.org
> http://mail.scipy.org/mailman/listinfo/numpy-discussion
>


More information about the NumPy-Discussion mailing list