Если у вас есть данные, сохраненные в массиве numpy
, вы можете использовать нарезку для извлечения необходимых вам подматриц. Вот пример с матрицей (3,5,3)
(но пример может быть применен к любому измерению):
A = numpy.array([[[1,1,1],
[2,2,2],
[3,3,3],
[4,4,4],
[5,5,5]],
[[11,11,11],
[21,21,21],
[31,31,31],
[41,41,41],
[51,51,51]],
[[12,12,12],
[22,22,22],
[32,32,32],
[42,42,42],
[52,52,52]]]
sub_matrix_1 = A[:,:,0]
print (sub_matrix_1)
Будет производить:
[[ 1 2 3 4 5]
[11 21 31 41 51]
[12 22 32 42 52]]
РЕДАКТИРОВАТЬ: также можно перебрать массив, чтобы получить массив 3-го измерения:
for i in range(A.shape[-1]):
print (A[:,:,i])
# Your submatrix is A[:,:,i], you can directly manipulate it