Numpy ndarray внутри массива с использованием индекса

Question

У меня есть numpy.ndarray - a

и a[0] дает мне:

array([0., 0., 0., ..., 0., 0., 0.])

также a[0:1] дает:

array([[0., 0., 0., ..., 0., 0., 0.]])

Как мне сделать a[0] в формате array([[0., 0., 0., ..., 0., 0., 0.]]) т.е. в формате массива внутри массива, как мы делаем [list[0]]

Answer 1

В случае, если вы хотите изменить форму массива, чтобы каждый элемент первой оси (каждая строка) представлял собой массив формы (1, m), вместо просто (m,) вы могли бы сделать следующее:

a = a[:,np.newaxis,:]

Таким образом, массив (n, m) преобразуется в форму (n, 1, m), и получение первой строки с a[0] возвращает:

array([[0., 0., 0., ..., 0., 0., 0.]])