Numpy 3D массив к списку - PullRequest
       13

Numpy 3D массив к списку

0 голосов
/ 17 марта 2019

У меня есть пустой массив, который выглядит следующим образом:

X=array([[[ 0.,  0.,  0.,  0.],
          [ 0.,  0.,  0.,  0.],
          [ 0.,  0.,  0.,  0.]],

        [[ 0.,  0.,  0.,  0.],
         [ 0.,  0.,  0.,  0.],
         [ 0.,  0.,  0.,  0.]]])

Мне очень легко составить список 2d-массивов вдоль оси = 0, выполнив

Xnew=list(X)
Xnew=[array([[ 0.,  0.,  0.,  0.],
             [ 0.,  0.,  0.,  0.],
             [ 0.,  0.,  0.,  0.]]),
      array([[ 0.,  0.,  0.,  0.],
             [ 0.,  0.,  0.,  0.],
             [ 0.,  0.,  0.,  0.]])]

Теперь я хочу создать тот же список по оси 1. Я хочу, чтобы мой вывод был

Xnew=[X[:,1,:],X[:,2,:], ...]

Как я могу легко и быстро выполнить эту операцию предпочтительно без использования циклов for?

...