Quick way to access first element in Numpy array with arbitrary number of dimensions?

Question

I have a function that I want to have quickly access the first (aka zeroth) element of a given Numpy array, which itself might have any number of dimensions. What\'s the qu

Answer 1

## y -- numpy array of shape (1, Ty)

if you want to get the first element:

use y.shape[0]

if you want to get the second element:

use y.shape[1]

Source: https://docs.scipy.org/doc/numpy/reference/generated/numpy.take.html

You can also use the take for more complicated extraction (to get few elements):

numpy.take(a, indices, axis=None, out=None, mode='raise')[source] Take elements from an array along an axis.