Numpy создать массив матриц

Question

Я пытаюсь хранить матрицы в массиве, однако, когда я добавляю матрицу, он будет получать каждый элемент и выводить только одномерный массив.

Пример кода:

matrix_array= np.array([]) 
for y in y_label: 
     matrix_array= np.append(matrix_array, np.identity(3)) 

Answer 1

Очевидно np.append неправильный инструмент для работы:

In [144]: np.append(np.array([]), np.identity(3)) 
Out[144]: array([ 1., 0., 0., 0., 1., 0., 0., 0., 1.]) 

С его документы:

Если axis не указан, values может быть любая форма и будет сплющена перед использованием.

С списком добавить

In [153]: alist=[] 
In [154]: for y in [1,2]: 
    ...:  alist.append(np.identity(3)) 
    ...:  
In [155]: alist 
Out[155]: 
[array([[ 1., 0., 0.], 
     [ 0., 1., 0.], 
     [ 0., 0., 1.]]), array([[ 1., 0., 0.], 
     [ 0., 1., 0.], 
     [ 0., 0., 1.]])] 
In [156]: np.array(alist) 
Out[156]: 
array([[[ 1., 0., 0.], 
     [ 0., 1., 0.], 
     [ 0., 0., 1.]], 

     [[ 1., 0., 0.], 
     [ 0., 1., 0.], 
     [ 0., 0., 1.]]]) 
In [157]: _.shape 
Out[157]: (2, 3, 3)